Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/kernels/
Dragged_gather_op_test.cc252 (*op.node_def.mutable_attr())["PARAMS_RAGGED_RANK"].set_i(1); in TEST_F()
253 (*op.node_def.mutable_attr())["OUTPUT_RAGGED_RANK"].set_i(1); in TEST_F()
262 (*op.node_def.mutable_attr())["PARAMS_RAGGED_RANK"].set_i(2); in TEST_F()
263 (*op.node_def.mutable_attr())["OUTPUT_RAGGED_RANK"].set_i(2); in TEST_F()
269 (*op.node_def.mutable_attr())["PARAMS_RAGGED_RANK"].set_i(1); in TEST_F()
270 (*op.node_def.mutable_attr())["OUTPUT_RAGGED_RANK"].set_i(2); in TEST_F()
277 (*op.node_def.mutable_attr())["PARAMS_RAGGED_RANK"].set_i(1); in TEST_F()
278 (*op.node_def.mutable_attr())["OUTPUT_RAGGED_RANK"].set_i(0); in TEST_F()
Dragged_tensor_to_variant_op_test.cc344 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(0); in TEST_F()
351 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(1); in TEST_F()
363 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(2); in TEST_F()
373 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(3); in TEST_F()
385 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(0); in TEST_F()
389 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(1); in TEST_F()
400 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(2); in TEST_F()
409 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(3); in TEST_F()
Dragged_tensor_to_sparse_kernel_test.cc146 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(0); in TEST_F()
150 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(1); in TEST_F()
161 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(2); in TEST_F()
170 (*op.node_def.mutable_attr())["RAGGED_RANK"].set_i(3); in TEST_F()
Dragged_tensor_from_variant_op_test.cc600 (*op.node_def.mutable_attr())["input_ragged_rank"].set_i(0); in TEST_F()
601 (*op.node_def.mutable_attr())["output_ragged_rank"].set_i(1); in TEST_F()
607 (*op.node_def.mutable_attr())["input_ragged_rank"].set_i(1); in TEST_F()
609 (*op.node_def.mutable_attr())["output_ragged_rank"].set_i(1); in TEST_F()
614 (*op.node_def.mutable_attr())["output_ragged_rank"].set_i(2); in TEST_F()
619 (*op.node_def.mutable_attr())["output_ragged_rank"].set_i(3); in TEST_F()
626 (*op.node_def.mutable_attr())["input_ragged_rank"].set_i(3); in TEST_F()
628 (*op.node_def.mutable_attr())["output_ragged_rank"].set_i(3); in TEST_F()
632 (*op.node_def.mutable_attr())["output_ragged_rank"].set_i(4); in TEST_F()
637 (*op.node_def.mutable_attr())["output_ragged_rank"].set_i(5); in TEST_F()
[all …]
Dragged_tensor_to_tensor_op_test.cc516 (*op_->node_def.mutable_attr())["num_row_partition_tensors"].set_i( in SetAttributes()
/external/llvm-project/polly/lib/External/isl/
Disl_pw_union_opt.c221 isl_set *better, *set_i, *set_j; in FN() local
231 set_i = isl_set_list_get_set(list1, i); in FN()
233 set_i = FN(PW,worse_or_out)(set_i, in FN()
235 list1 = isl_set_list_set_set(list1, i, set_i); in FN()
236 set_i = isl_set_copy(pw1->p[i].set); in FN()
238 set_j = FN(PW,better_or_out)(set_j, better, set_i); in FN()
Disl_coalesce.c1755 struct isl_set *set_i = NULL; in can_wrap_in_facet() local
1762 set_i = set_from_updated_bmap(info[i].bmap, info[i].tab); in can_wrap_in_facet()
1771 if (!set_i || !set_j || !bound) in can_wrap_in_facet()
1781 if (add_wraps(&wraps, &info[j], bound->el, set_i) < 0) in can_wrap_in_facet()
1799 isl_set_free(set_i); in can_wrap_in_facet()
1808 isl_set_free(set_i); in can_wrap_in_facet()
1835 struct isl_coalesce_info *info_j, __isl_keep isl_set *set_i, in wrap_in_facet() argument
1846 else if (add_wraps(wraps, info_j, wraps->mat->row[w], set_i) < 0) in wrap_in_facet()
1874 __isl_keep isl_set *set_i) in try_wrap_in_facets() argument
1897 if (wrap_in_facet(wraps, w, &info[j], set_i, snap) < 0) in try_wrap_in_facets()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.cc50 device_ordinal_value.set_i(device_ordinal); in SetDeviceOrdinalAttributeForNode()
61 device_ordinal_value.set_i(device_ordinal); in SetDeviceOrdinalAttributeForNode()
Dsharding_util_test.cc70 index.set_i(0); in TEST_P()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc586 (*op->mutable_attr())["block_size"].set_i(src_op.block_size); in ConvertDepthToSpaceOperator()
597 (*op->mutable_attr())["block_size"].set_i(src_op.block_size); in ConvertSpaceToDepthOperator()
700 (*add_op->mutable_attr())["N"].set_i(src_op.inputs.size()); in ConvertAddNOperator()
951 (*lrn_op->mutable_attr())["depth_radius"].set_i(src_op.range); in ConvertLocalResponseNormalizationOperator()
968 (*fakequant_op->mutable_attr())["num_bits"].set_i(src_op.num_bits); in ConvertFakeQuantOperator()
1046 (*dc_op->mutable_attr())["N"].set_i(src_op.inputs.size()); in ConvertConcatenationOperator()
1158 (*split_op->mutable_attr())["num_split"].set_i(src_op.num_split); in ConvertSplitOperator()
1183 (*split_v_op->mutable_attr())["num_split"].set_i(src_op.num_split); in ConvertSplitVOperator()
1354 (*pack_op->mutable_attr())["axis"].set_i(src_op.axis); in ConvertPackOperator()
1355 (*pack_op->mutable_attr())["N"].set_i(src_op.inputs.size()); in ConvertPackOperator()
[all …]
/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/core/grappler/optimizers/data/
Denable_gradient_descent.cc61 (*model_node.mutable_attr())[kAlgorithm].set_i(GRADIENT_DESCENT); in OptimizeAndCollectStats()
Dslack.cc84 (*dataset_node->mutable_attr())["slack_period"].set_i(slack_period_); in RecursivelyHandleOp()
Dautotune_buffer_sizes.cc81 node.mutable_attr()->at(kBufferSizeMin).set_i(initial_buffer_size); in OptimizeAndCollectStats()
Dhoist_random_uniform.cc126 (*zip_node.mutable_attr())["N"].set_i(2); in MakeZipNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc787 list->set_i(2, 0); in UpdateAttrExplicitPaddings()
788 list->set_i(3, 0); in UpdateAttrExplicitPaddings()
789 list->set_i(4, height_before); in UpdateAttrExplicitPaddings()
790 list->set_i(5, height_after); in UpdateAttrExplicitPaddings()
791 list->set_i(6, width_before); in UpdateAttrExplicitPaddings()
792 list->set_i(7, width_after); in UpdateAttrExplicitPaddings()
992 list->set_i(1, c); in UpdateTuple()
993 list->set_i(2, h); in UpdateTuple()
994 list->set_i(3, w); in UpdateTuple()
1839 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()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dauto_shard_dataset_op.cc92 attr.set_i(pair.second); in CreateConfig()
/external/protobuf/src/google/protobuf/
Dwire_format_unittest.cc428 ->set_i(123); in TEST()
475 ->set_i(123); in TEST()
531 message.set_i(123); in TEST()
586 message.set_i(123); in TEST()
676 m->set_i(i); in TEST()
697 message.mutable_a()->mutable_a()->mutable_a()->mutable_a()->set_i(1); in TEST()
/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h107 attr.set_i(num_nodes); in CreateGraphDef()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc344 device_ordinal_temp_value.set_i(0); in TEST_F()
500 device_ordinal_temp_value.set_i(0); in TEST_F()
650 device_ordinal_temp_value.set_i(0); in TEST_F()
786 device_ordinal_temp_value.set_i(0); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc340 (*node.mutable_attr())["index"].set_i(index - 1); in ReplaceInputWithConst()
394 (*node.mutable_attr())["index"].set_i(to); in RemoveFunctionOutputs()
Dgraph_view_internal_test.cc409 attr_3.set_i(4); in TEST()
1053 attr_1.set_i(8); in TEST()
1075 attr_1.set_i(8); in TEST()
/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.cc168 attr->mutable_default_value()->set_i(17); in TEST()

123