Home
last modified time | relevance | path

Searched refs:FillIota (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_test.cc53 test::FillIota<float>(&a_data, 1.0f); in TestSimpleAdd()
58 test::FillIota<float>(&b_data, 1.0f); in TestSimpleAdd()
74 test::FillIota<float>(&placeholder_tensor, 1.0f); in TestSimpleAdd()
90 test::FillIota<float>(&a_data, 1.0f); in TestOpExclusionAdd()
95 test::FillIota<float>(&b_data, 1.0f); in TestOpExclusionAdd()
111 test::FillIota<float>(&placeholder_tensor, 1.0f); in TestOpExclusionAdd()
135 test::FillIota<float>(&placeholder_tensor, 1.0); in TestShapePropagation()
164 test::FillIota<float>(&placeholder_tensor, 1.0); in TestPreserveOutputShapes()
227 test::FillIota<float>(&a_const_data, 1.0f); in TestReplaceSendRecvs()
301 test::FillIota<float>(&a_data, 1.0f); in TestRemoveUnusedNodes()
[all …]
Dtransform_utils_test.cc39 test::FillIota<float>(&a_data, 1.0f); in TestMapNamesToNodes()
43 test::FillIota<float>(&b_data, 1.0f); in TestMapNamesToNodes()
72 test::FillIota<float>(&a_data, 1.0f); in TestMapNodesToOutputs()
76 test::FillIota<float>(&b_data, 1.0f); in TestMapNodesToOutputs()
200 test::FillIota<int32>(&input_tensor, 1); in TestSetNodeTensorAttrWithTensor()
211 test::FillIota<int32>(&input_tensor, 1); in TestGetNodeTensorAttr()
226 test::FillIota<float>(&a_data, 1.0f); in TestFilterGraphDef()
230 test::FillIota<float>(&b_data, 1.0f); in TestFilterGraphDef()
286 test::FillIota<float>(&a_data, 1.0f); in TestGetOpTypeMatches()
290 test::FillIota<float>(&b_data, 1.0f); in TestGetOpTypeMatches()
[all …]
Dquantize_nodes_test.cc349 test::FillIota<float>(&input_tensor, 1); in TestQuantizeBiasAdd()
354 test::FillIota<float>(&offset_tensor, 1); in TestQuantizeBiasAdd()
826 test::FillIota<float>(&a_data, 1.0f); in TestInputRange()
838 test::FillIota<float>(&placeholder_tensor, 1.0f); in TestInputRange()
851 test::FillIota<float>(&a_data, 1.0f); in TestFallbackRange()
863 test::FillIota<float>(&placeholder_tensor, 1.0f); in TestFallbackRange()
880 test::FillIota<float>(&input_tensor, 1); in TestConvertFakeQuantsToRequantize()
885 test::FillIota<float>(&offset_tensor, 1); in TestConvertFakeQuantsToRequantize()
1032 test::FillIota<float>(&input_tensor, 1); in TestConvertFakeQuantsEndToEnd()
1037 test::FillIota<float>(&offset_tensor, 1); in TestConvertFakeQuantsEndToEnd()
[all …]
Dinsert_logging_test.cc50 test::FillIota<float>(&const_tensor, 1.0f); in TestInsertLogging()
97 test::FillIota<float>(&const_tensor, 1.0f); in TestInsertLoggingByOpType()
146 test::FillIota<float>(&const_tensor, 1.0f); in TestInsertLoggingByPrefix()
Dtransform_graph_test.cc59 test::FillIota<float>(&a_data, 1.0f); in TestConstantFolding()
64 test::FillIota<float>(&b_data, 1.0f); in TestConstantFolding()
Dfile_utils_test.cc39 test::FillIota<float>(&a_data, 1.0f); in TestLoadTextOrBinaryGraphFile()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test_utils.cc60 test::FillIota<float>(&a_data, 1.0f); in BuildMultipleAddGraph()
64 test::FillIota<float>(&b_data, 1.0f); in BuildMultipleAddGraph()
68 test::FillIota<float>(&c_data, 1.0f); in BuildMultipleAddGraph()
72 test::FillIota<float>(&d_data, 1.0f); in BuildMultipleAddGraph()
76 test::FillIota<float>(&e_data, 1.0f); in BuildMultipleAddGraph()
Dmfcc_op_test.cc41 test::FillIota<float>(&spectrogram_tensor, 1.0f); in TEST()
Dconv_ops_test.cc194 test::FillIota<float>(&input_data, 1.0f); in CompareFusedAndSeparate()
201 test::FillIota<float>(&filter_data, 1.0f); in CompareFusedAndSeparate()
255 test::FillIota<float>(&input_data, 1.0f); in CompareFusedPadOnlyAndSeparate()
262 test::FillIota<float>(&filter_data, 1.0f); in CompareFusedPadOnlyAndSeparate()
Dnn_ops_test.cc772 test::FillIota<float>(&input, 1.0); in BM_LRNFloat()
852 test::FillIota<float>(&input1, 1.0); in BM_AvgPool()
956 test::FillIota<float>(&output_backprop, 11.0); in BM_AvgPoolBk()
1047 test::FillIota<float>(&input1, 1.0); in BM_MaxPool()
1226 test::FillIota<float>(&input1, 1.0); in BM_ReluFloat()
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc69 test::FillIota<float>(&dy, 0); in TEST()
109 test::FillIota<float>(&dy0, 0); in TEST()
110 test::FillIota<float>(&dy1, 100); in TEST()
171 test::FillIota<float>(&dy, 0); in TEST()
279 test::FillIota<float>(&dy0, 0); in TEST()
280 test::FillIota<float>(&dy1, 100); in TEST()
334 test::FillIota<float>(&dy, 73); in TEST()
373 test::FillIota<float>(&dy, 73); in TEST()
408 test::FillIota<float>(&dy, 1); in TEST()
441 test::FillIota<float>(&dy, 0); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix_test.cc65 input.FillIota(0); in XLA_TEST_F()
80 input.FillIota(0); in TestMatrixDiagonal()
94 input.FillIota(0); in TestSetMatrixDiagonal()
130 input.FillIota(0); in TestMatrixDiagonal4D()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_rewriter_transform_test.cc46 test::FillIota<float>(&a_data, 1.0f); in TEST()
50 test::FillIota<float>(&b_data, 1.0f); in TEST()
Dgraph_transferer_test.cc157 test::FillIota<float>(&input_data, 1.0f); in CreateConvGraphDef()
161 test::FillIota<float>(&filter_data, 1.0f); in CreateConvGraphDef()
176 test::FillIota<float>(&input_data, 1.0f); in CreatePoolGraphDef()
180 test::FillIota<float>(&filter_data, 1.0f); in CreatePoolGraphDef()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc96 array.FillIota(1.0f); in TEST_F()
113 array.FillIota(1.0f); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dslice_test.cc43 values.FillIota(0); in TEST_F()
56 values.FillIota(0); in TEST_F()
69 values.FillIota(0); in TEST_F()
171 values.FillIota(1); in TEST_F()
181 values.FillIota(1); in TEST_F()
495 values.FillIota(3.14159); in Run()
Ddot_operation_test.cc1688 input_arr.FillIota(0); in XLA_TEST_F()
1689 const_arr.FillIota(0); in XLA_TEST_F()
1705 input_arr.FillIota(0); in XLA_TEST_F()
1706 const_arr.FillIota(0); in XLA_TEST_F()
1726 input_arr.FillIota(0); in XLA_TEST_F()
1727 const_arr.FillIota(0); in XLA_TEST_F()
1743 input_arr.FillIota(0); in XLA_TEST_F()
1744 const_arr.FillIota(0); in XLA_TEST_F()
1767 input_arr.FillIota(0); in XLA_TEST_F()
1768 const_arr.FillIota(0); in XLA_TEST_F()
[all …]
Ddynamic_ops_test.cc522 input_values.FillIota(static_cast<T>(0)); in RunR3Contiguous()
524 update_values.FillIota(static_cast<T>(value)); in RunR3Contiguous()
528 expected_values.FillIota(static_cast<T>(0)); in RunR3Contiguous()
Dtriangular_solve_test.cc420 bvals.FillIota(1.); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc163 input.FillIota(0); in TEST_F()
176 input.FillIota(0); in TEST_F()
189 input.FillIota(0); in TEST_F()
202 input.FillIota(0); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_test.cc96 test::FillIota<T>(&input_data, static_cast<T>(1)); in SimpleConv2D()
103 test::FillIota<T>(&filter_data, static_cast<T>(1)); in SimpleConv2D()
143 test::FillIota<float>(&output_data, 1.0f); in SimpleConv2DBackpropInput()
531 test::FillIota<float>(&bias_data, 1.0f); in TEST_F()
Dlayout_optimizer_test.cc78 test::FillIota<T>(&input_data, static_cast<T>(1)); in SimpleConv2D()
85 test::FillIota<T>(&filter_data, static_cast<T>(1)); in SimpleConv2D()
152 test::FillIota<float>(&output_data, 1.0f); in SimpleConv2DBackpropInput()
203 test::FillIota<float>(&data, 1.0f); in SimpleFusedBatchNormGrad()
210 test::FillIota<float>(&data_vector, 2.0f); in SimpleFusedBatchNormGrad()
/external/tensorflow/tensorflow/core/framework/
Dtensor_testutil.h82 void FillIota(Tensor* tensor, const T& val) { in FillIota() function
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration_test.cc217 test::FillIota<float>(&expected_default, 33.0); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util_test.cc83 tile.FillIota(0); in TEST()

12