Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
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
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
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
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
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
/external/tensorflow/tensorflow/dtensor/mlir/
Dundo_merge_const_across_mesh.cc44 getOperation().walk([&builder](mlir::TF::ConstOp const_op) { in runOnOperation()
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
Dsparse_expansion.cc95 auto const_op = op.getOperand(3).getDefiningOp(); in RemoveUnusedSparseToDenseOps() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dunfold_large_splat_constant.cc55 module.walk([&](mlir::arith::ConstantOp const_op) { in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID()
Ddilated_conv.h183 if (auto const_op = in matchAndRewrite() local
Dprepare_quantize_helper.h280 SourceOp op, Operation* const_op, int input_index, in processConstantOp()
/external/tensorflow/tensorflow/core/transforms/constant_folding/
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 …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dsink_constant.cc60 auto const_op = dyn_cast_or_null<TF::ConstOp>(constant.getDefiningOp()); in runOnOperation() local
Dfreeze_saved_model_assets.cc94 auto const_op = builder.create<TF::ConstOp>( in runOnOperation() local
Dinitialize_variables_in_session_init.cc49 auto const_op = builder.create<mlir::arith::ConstantOp>( in InitializeVariable() local
Dfreeze_global_tensors.cc207 auto const_op = builder.create<TF::ConstOp>(global_tensor.getLoc(), in runOnOperation() local
Dreplicate_to_island.cc112 auto const_op = builder.create<TF::ConstOp>( in UpdateRegionReplicateVariantOps() local
Dconvert_control_to_data_outputs.cc331 auto const_op = builder.create<TF::ConstOp>(loc, val); in GetDummyConstant() local
Dcollection_ops_util.cc248 auto const_op = llvm::dyn_cast<TF::ConstOp>(indices_op); in GetFirstIfIndicesAreContiguous() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_testutils.h59 auto const_op = ops::Const(s.WithOpName(name), t); in MakeConstNodeDef() local
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dnullary_spmd_expander.cc55 if (auto const_op = mlir::dyn_cast<mlir::TF::ConstOp>(op)) { in ExpandOp() local
/external/tensorflow/tensorflow/lite/toco/
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
/external/tensorflow/tensorflow/core/ir/
Dtf_op_wrapper_test.cc176 TFOp const_op = &(*iterator++); in TEST() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond_test.cc123 auto const_op = ops::Const(root.WithOpName("const"), 1); in TEST() local
/external/deqp-deps/SPIRV-Tools/source/opt/
Dfold.cpp640 const analysis::Constant* const_op = const_mgr->FindDeclaredConstant(id); in FoldInstructionToConstant() local
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dfold.cpp640 const analysis::Constant* const_op = const_mgr->FindDeclaredConstant(id); in FoldInstructionToConstant() local
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dfold.cpp640 const analysis::Constant* const_op = const_mgr->FindDeclaredConstant(id); in FoldInstructionToConstant() local

12