Home
last modified time | relevance | path

Searched defs:const_op (Results 1 – 25 of 38) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_utils.cc191 auto const_op = in getTosaConst8bitTable() local
234 auto const_op = in getTosaConst16bitTable() local
313 auto const_op = in getTosaConstTensorSingleF32() local
324 auto const_op = in getTosaConstTensorSingleI32() local
495 auto const_op = in getConstTensor() local
519 auto const_op = in getConstTensor() local
542 auto const_op = in getConstTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc227 FailureOr<TFOp> const_op = CreateConstantTensorOp( in ReplaceOpWithConstantTensor() local
323 TFOp const_op = *CreateConstantTensorOp( in ReplaceOperationWithBroadcastTo() local
736 FailureOr<TFOp> const_op = CreateConstantTensorOp( in matchAndRewrite() local
813 TFOp const_op = *CreateConstantTensorOp( in matchAndRewrite() local
848 TFOp const_op = *CreateConstantTensorOp( in matchAndRewrite() local
882 TFOp const_op = *CreateConstantTensorOp( in matchAndRewrite() local
925 TFOp const_op = *CreateConstantTensorOp( in matchAndRewrite() local
955 FailureOr<TFOp> const_op = CreateConstantTensorOp( in matchAndRewrite() local
1066 FailureOr<TFOp> const_op = CreateConstantTensorOp( in matchAndRewrite() local
1158 FailureOr<TFOp> const_op = CreateConstantTensorOp( in matchAndRewrite() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dexport_tensorflow.cc167 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertFloatTensorConst() local
209 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertFloatTensorConst() local
232 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertBoolTensorConst() local
256 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertIntTensorConst() local
280 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in CreateIntTensorConst() local
305 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertComplex64TensorConst() local
328 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in CreateMatrixShapeTensorConst() local
346 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in CreateDummyConcatDimTensorConst() local
361 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in CreateReshapeShapeTensorConst() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Dfold_constants_to_subgraph.cc88 void CopyConstantIntoFunc(int argument_index, Operation* const_op, in CopyConstantIntoFunc()
104 if (auto const_op = dyn_cast_or_null<TFL::ConstOp>(op)) { in IsConstOrQConstInt() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dremove_tf_if_const_args.cc66 if (auto const_op = operand.getDefiningOp<mlir::TF::ConstOp>()) { in ProcessFunction() local
136 auto const_op = in CreateBranchFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dhandle_cross_cluster_dependencies.cc83 mlir::LogicalResult CloneOpToCluster(mlir::Operation* const_op, in CloneOpToCluster()
170 mlir::Operation* const_op = GetConstOp(input_value.getDefiningOp()); in CloneConstantsAcrossMesh() local
H A Dundo_merge_const_across_mesh.cc44 getOperation().walk([&builder](mlir::TF::ConstOp const_op) { in runOnOperation()
H A Dsparse_expansion.cc95 auto const_op = op.getOperand(3).getDefiningOp(); in RemoveUnusedSparseToDenseOps() local
H A Dlayout_propagation_v2.cc1221 [&](mlir::TF::ConstOp const_op) { const_ops.emplace_back(const_op); }); in DuplicateConstants()
1223 for (mlir::TF::ConstOp const_op : const_ops) { in DuplicateConstants() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dunfold_large_splat_constant.cc55 module.walk([&](mlir::arith::ConstantOp const_op) { in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID()
H A Ddilated_conv.h183 if (auto const_op = in matchAndRewrite() local
H A Dprepare_quantize_helper.h280 SourceOp op, Operation* const_op, int input_index, in processConstantOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dsink_constant.cc60 auto const_op = dyn_cast_or_null<TF::ConstOp>(constant.getDefiningOp()); in runOnOperation() local
H A Dfreeze_saved_model_assets.cc94 auto const_op = builder.create<TF::ConstOp>( in runOnOperation() local
H A Dinitialize_variables_in_session_init.cc49 auto const_op = builder.create<mlir::arith::ConstantOp>( in InitializeVariable() local
H A Dfreeze_global_tensors.cc207 auto const_op = builder.create<TF::ConstOp>(global_tensor.getLoc(), in runOnOperation() local
H A Dreplicate_to_island.cc112 auto const_op = builder.create<TF::ConstOp>( in UpdateRegionReplicateVariantOps() local
H A Dconvert_control_to_data_outputs.cc331 auto const_op = builder.create<TF::ConstOp>(loc, val); in GetDummyConstant() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
H A Dtrt_testutils.h59 auto const_op = ops::Const(s.WithOpName(name), t); in MakeConstNodeDef() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dnullary_spmd_expander.cc55 if (auto const_op = mlir::dyn_cast<mlir::TF::ConstOp>(op)) { in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/
H A Dtf_op_wrapper_test.cc176 TFOp const_op = &(*iterator++); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfunctionalize_cond_test.cc123 auto const_op = ops::Const(root.WithOpName("const"), 1); in TEST() local
/aosp_15_r20/external/deqp-deps/SPIRV-Tools/source/opt/
Dfold.cpp588 const analysis::Constant* const_op = const_mgr->FindDeclaredConstant(id); in FoldInstructionToConstant() local
/aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/source/opt/
H A Dfold.cpp588 const analysis::Constant* const_op = const_mgr->FindDeclaredConstant(id); in FoldInstructionToConstant() local
/aosp_15_r20/external/angle/third_party/spirv-tools/src/source/opt/
H A Dfold.cpp588 const analysis::Constant* const_op = const_mgr->FindDeclaredConstant(id); in FoldInstructionToConstant() local

12