Home
last modified time | relevance | path

Searched defs:add_op (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/python/kernel_tests/variables/
Ddense_update_ops_no_tsan_test.py45 def run_add(add_op): argument
111 def run_add(add_op): argument
/external/tensorflow/tensorflow/dtensor/mlir/
Ddtensor_allreduce_sum_optimization.cc69 mlir::Operation* add_op, in CheckReduceAndSumOptimizationCriteria()
264 mlir::Operation** add_op, mlir::TF::DTensorAllReduceOp* all_reduce_op, in CheckWhileLoopOptimizationCriteria()
369 mlir::Operation* add_op, bool* changed) { in ExtractAllReduceFromWhileOp()
426 mlir::Operation* add_op = nullptr; in OptimizeWhileLoopLazyAllReduce() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/python/
Dsmoketest.py36 add_op = m.body.operations[0].regions[0].blocks[0].operations[0] variable
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dremove_successive_transpose_test.cc68 auto* add_op = new toco::AddOperator; in CreateGraph() local
126 auto* add_op = new toco::AddOperator; in TEST_F() local
Dfuse_binary_into_preceding_affine_test.cc93 auto* add_op = new AddOperator; in TEST_F() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_trivial_addn_to_add.cc42 auto* add_op = new AddOperator; in Run() local
Didentify_prelu.cc50 const auto* add_op = add_op_it->get(); in Run() local
Didentify_l2_normalization.cc63 Operator* add_op = nullptr; in Run() local
Dresolve_batch_normalization.cc64 auto* add_op = new AddOperator; in Run() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgpu_fusion.cc71 AddV2Op add_op; in matchAndRewrite() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_test.cc366 auto* add_op = TF_NewAbstractOp(graph_ctx); in TEST_P() local
565 auto* add_op = TF_NewAbstractOp(graph_ctx); in TEST_P() local
587 auto* add_op = TF_NewAbstractOp(graph_ctx); in TEST_P() local
721 auto* add_op = TF_NewAbstractOp(graph_ctx); in TEST_P() local
745 auto* add_op = TF_NewAbstractOp(graph_ctx); in TEST_P() local
/external/tensorflow/tensorflow/core/ir/
Dtf_op_wrapper_test.cc177 TFOp add_op = &(*iterator); in TEST() local
/external/tensorflow/tensorflow/core/tfrt/eager/
Dfunction_cache_test.cc63 tensorflow::AbstractOperationPtr add_op(ctx->CreateOperation()); in Add() local
/external/XNNPACK/src/operators/
Dbinary-elementwise-nd.c1105 xnn_operator_t add_op, in xnn_setup_add_nd_f16()
1125 xnn_operator_t add_op, in xnn_setup_add_nd_f32()
1145 xnn_operator_t add_op, in xnn_setup_add_nd_qs8()
1168 xnn_operator_t add_op, in xnn_setup_add_nd_qu8()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc1212 Output add_op = in TestMergeDuplicateQuantizes() local
1243 Output add_op = Add(root.WithOpName("add_op"), a_op, b_op); in TestMergeDuplicateConsts() local
1292 Output add_op = Add(root.WithOpName("add_op"), a_relu_op, b_relu_op); in TestMergeDuplicatesNested() local
1342 Output add_op = Add(root.WithOpName("add_op"), a_relu_op, b_relu_op); in TestMergeDuplicatesInOut() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmkl_remapper_test.cc39 const string& activation, string add_op, in FuseConv2DWithBiasAndAddNOrAdd()
541 void RelpaceAddWithBiasAddDepthwiseConv2D(const string& add_op) { in RelpaceAddWithBiasAddDepthwiseConv2D()
/external/mesa3d/src/compiler/nir/
Dnir_loop_analyze.c664 nir_op add_op; in will_break_on_first_iteration() local
703 nir_op add_op; in test_iterations() local
/external/harfbuzz_ng/src/
Dhb-cff-interp-common.hh528 void add_op (op_code_t op, const byte_str_ref_t& str_ref = byte_str_ref_t ()) in add_op() function
538 void add_op (op_code_t op, const byte_str_ref_t& str_ref, const VAL &v) in add_op() function
Dhb-subset-cff-common.hh317 void add_op (op_code_t op, const byte_str_ref_t& str_ref) in add_op() function
/external/mesa3d/src/gallium/drivers/vc4/
Dvc4_qpu_schedule.c326 uint32_t add_op = QPU_GET_FIELD(inst, QPU_OP_ADD); in calculate_deps() local
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_events_reader.py525 def add_op(self, graph_op_creation_digest): member in DebuggedGraph
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc687 tensorflow::NodeDef* add_op = tensorflow_graph->add_node(); in ConvertAddOperator() local
699 tensorflow::NodeDef* add_op = tensorflow_graph->add_node(); in ConvertAddNOperator() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc666 auto add_op = dyn_cast_or_null<TFL::AddOp>(fc_op.input().getDefiningOp()); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1403 auto add_op = CreateOpAndInfer<tosa::AddOp>( in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2270 auto add_op = rewriter.create<chlo::BroadcastAddOp>( in matchAndRewrite() local
3063 auto add_op = rewriter.create<AddOp>(loc, mul_op, half); in matchAndRewrite() local
4564 auto add_op = builder.create<AddOp>(loc, block->getArgument(0), in BuildScatterBody() local

12