Home
last modified time | relevance | path

Searched refs:set_i (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dragged_gather_op_test.cc250 (*op.node_def.mutable_attr())["PARAMS_RAGGED_RANK"].set_i(1); in TEST_F()
251 (*op.node_def.mutable_attr())["OUTPUT_RAGGED_RANK"].set_i(1); in TEST_F()
260 (*op.node_def.mutable_attr())["PARAMS_RAGGED_RANK"].set_i(2); in TEST_F()
261 (*op.node_def.mutable_attr())["OUTPUT_RAGGED_RANK"].set_i(2); in TEST_F()
267 (*op.node_def.mutable_attr())["PARAMS_RAGGED_RANK"].set_i(1); in TEST_F()
268 (*op.node_def.mutable_attr())["OUTPUT_RAGGED_RANK"].set_i(2); in TEST_F()
275 (*op.node_def.mutable_attr())["PARAMS_RAGGED_RANK"].set_i(1); in TEST_F()
276 (*op.node_def.mutable_attr())["OUTPUT_RAGGED_RANK"].set_i(0); in TEST_F()
Dragged_tensor_to_sparse_kernel_test.cc145 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(0); in TEST_F()
149 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(1); in TEST_F()
160 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(2); in TEST_F()
169 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(3); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dauto_shard_dataset_op.cc93 num_workers_attr.set_i(num_workers_); in CreateGrapplerRewriteConfig()
98 index_attr.set_i(index_); in CreateGrapplerRewriteConfig()
Drebatch_dataset_op.cc82 num_workers_attr.set_i(num_workers_); in CreateGrapplerRewriteConfig()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.cc42 device_ordinal_value.set_i(device_ordinal); in SetDeviceOrdinalAttributeForNode()
53 device_ordinal_value.set_i(device_ordinal); in SetDeviceOrdinalAttributeForNode()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc581 (*op->mutable_attr())["block_size"].set_i(src_op.block_size); in ConvertDepthToSpaceOperator()
592 (*op->mutable_attr())["block_size"].set_i(src_op.block_size); in ConvertSpaceToDepthOperator()
695 (*add_op->mutable_attr())["N"].set_i(src_op.inputs.size()); in ConvertAddNOperator()
943 (*lrn_op->mutable_attr())["depth_radius"].set_i(src_op.range); in ConvertLocalResponseNormalizationOperator()
960 (*fakequant_op->mutable_attr())["num_bits"].set_i(src_op.num_bits); in ConvertFakeQuantOperator()
1038 (*dc_op->mutable_attr())["N"].set_i(src_op.inputs.size()); in ConvertConcatenationOperator()
1149 (*split_op->mutable_attr())["num_split"].set_i(src_op.num_split); in ConvertSplitOperator()
1172 (*split_v_op->mutable_attr())["num_split"].set_i(src_op.num_split); in ConvertSplitVOperator()
1341 (*pack_op->mutable_attr())["axis"].set_i(src_op.axis); in ConvertPackOperator()
1342 (*pack_op->mutable_attr())["N"].set_i(src_op.inputs.size()); in ConvertPackOperator()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc766 list->set_i(2, 0); in UpdateAttrExplicitPaddings()
767 list->set_i(3, 0); in UpdateAttrExplicitPaddings()
768 list->set_i(4, height_before); in UpdateAttrExplicitPaddings()
769 list->set_i(5, height_after); in UpdateAttrExplicitPaddings()
770 list->set_i(6, width_before); in UpdateAttrExplicitPaddings()
771 list->set_i(7, width_after); in UpdateAttrExplicitPaddings()
941 list->set_i(1, c); in UpdateTuple()
942 list->set_i(2, h); in UpdateTuple()
943 list->set_i(3, w); in UpdateTuple()
1780 node_->mutable_attr()->at(mask).set_i(i); in UpdateMask()
[all …]
Dmemory_optimizer_test.cc57 .set_i(0); in TEST_F()
97 .set_i(0); in TEST_F()
120 .set_i(0); in TEST_F()
122 .set_i(0); in TEST_F()
Dconstant_folding.cc3139 (*node->mutable_attr())["N"].set_i(node->input_size() - in PartialAssocOpConstFolding()
3142 (*added_node->mutable_attr())["N"].set_i(const_inputs.size()); in PartialAssocOpConstFolding()
3232 (*added_node->mutable_attr())["N"].set_i(interval.second - in PartialConcatConstFolding()
3251 (*node->mutable_attr())["N"].set_i(node->input_size() - 1); in PartialConcatConstFolding()
3304 (*parent->mutable_attr())["N"].set_i(NumNonControlInputs(*parent) - 1); in MergeConcat()
Dloop_optimizer.cc245 parallel_iterations.set_i(piterations); in HandleInvariantNode()
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration_test.cc84 (*mutable_attr)["Nsparse"].set_i(3); in TEST_F()
99 (*mutable_attr)["Ndense"].set_i(2); in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc348 device_ordinal_temp_value.set_i(0); in TEST_F()
433 device_ordinal_temp_value.set_i(0); in TEST_F()
494 device_ordinal_temp_value.set_i(0); in TEST_F()
605 device_ordinal_temp_value.set_i(0); in TEST_F()
764 device_ordinal_temp_value.set_i(0); in TEST_F()
904 device_ordinal_temp_value.set_i(0); in TEST_F()
Dextract_outside_compilation_pass.cc493 device_ordinal_attr.set_i(0); in ConstructHostGraph()
613 device_ordinal_attr.set_i(0); in ExpandHostGraphIntoMainGraph()
690 device_ordinal_attr.set_i(0); in RewriteShapeInferenceGraph()
830 device_ordinal_attr.set_i(0); in ReplaceKeyPlaceholderWithArgNode()
1032 device_ordinal_temp_value.set_i(0); in RewriteHostWhileLoopCond()
1112 device_ordinal_temp_value.set_i(0); in RewriteHostWhileLoopBody()
/external/protobuf/src/google/protobuf/
Dwire_format_unittest.cc427 unittest::TestMessageSetExtension1::message_set_extension)->set_i(123); in TEST()
470 unittest::TestMessageSetExtension1::message_set_extension)->set_i(123); in TEST()
522 message.set_i(123); in TEST()
570 message.set_i(123); in TEST()
616 message.mutable_a()->mutable_a()->mutable_a()->mutable_a()->set_i(1); in TEST()
/external/tensorflow/tensorflow/core/ops/
Dtpu_embedding_ops.cc108 table_id_attr->mutable_default_value()->set_i(-1); in RegisterPerTableLoadOpsForAlgorithmBody()
225 table_id_attr->mutable_default_value()->set_i(-1); in RegisterPerTableRetrieveOpsForAlgorithmBody()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc159 parameters["minimum_segment_size"].set_i(minimum_segment_size); in ConvertGraphDefToTensorRT()
160 parameters["max_batch_size"].set_i(max_batch_size); in ConvertGraphDefToTensorRT()
162 parameters["max_workspace_size_bytes"].set_i(max_workspace_size_bytes); in ConvertGraphDefToTensorRT()
165 parameters["maximum_cached_engines"].set_i(max_cached_engines); in ConvertGraphDefToTensorRT()
/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h107 attr.set_i(num_nodes); in CreateGraphDef()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dhoist_random_uniform.cc126 (*zip_node.mutable_attr())["N"].set_i(2); in MakeZipNode()
/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_svdf.cc142 (*svdf_node->mutable_attr())["Rank"].set_i(rank); in CreateNodes()
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util_test.cc157 attr->mutable_default_value()->set_i(17); in TEST()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc142 attr.set_i(stat.length); in ExtractExtraProperties()
/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc366 (*node.mutable_attr())["apple"].set_i(7); in TEST()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc271 (*m->mutable_attr())["N"].set_i(2); in RunInfiniteTFLoop()
/external/clang/utils/TableGen/
DClangAttrEmitter.cpp722 auto set_i = unique_set.find(i); in uniqueEnumsInOrder() local
723 if (set_i != unique_set.end()) { in uniqueEnumsInOrder()
725 unique_set.erase(set_i); in uniqueEnumsInOrder()
/external/protobuf/src/google/protobuf/compiler/cpp/
Dcpp_unittest.cc841 message.set_i(1); in TEST()
853 message.mutable_optional_message()->set_i(1); in TEST()

12