| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops_test.cc | 77 test::FillFn<float>(&data, [](const int i) { return 1.0f; }); in Ones()
83 test::FillFn<int64>(&data, [&](const int i) { return i; }); in SparseIndices()
91 test::FillFn<int64>(&data, [&](const int i) { return i / 4; }); in SparseExampleIndices()
99 test::FillFn<int64>( in SparseFeatureIndices()
106 test::FillFn<float>(&data, [](const int i) { in RandomZeroOrOne()
115 test::FillFn<float>(&data, [](const int i) { in RandomZeroOrOneMatrix()
|
| D | requantization_range_op_test.cc | 74 test::FillFn<qint32>(&quantized_tensor, [](int n) { return qint32(n); }); in BM_RequantizationRange()
|
| D | ops_testutil.h | 100 test::FillFn(AddInput(DataTypeToEnum<T>::v(), shape), input_mapping); in AddInput()
|
| D | segment_reduction_ops_test.cc | 56 test::FillFn<Index>(&input2, [&num_rows, &segment_size](Index i) -> Index { in BM_SegmentReduction()
|
| D | restore_v2_op_test.cc | 46 test::FillFn(&input, input_mapping); in MakeInput()
|
| D | restore_op_test.cc | 60 test::FillFn(&input, input_mapping); in MakeInput()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 110 test::FillFn<Scalar>(&test_case->expected, in MakeReductionTestCase()
113 test::FillFn<Scalar>(&test_case->expected, in MakeReductionTestCase()
116 test::FillFn<Scalar>(&test_case->expected, [](int) { return -max_; }); in MakeReductionTestCase()
118 test::FillFn<Scalar>(&test_case->expected, [](int) { return max_; }); in MakeReductionTestCase()
130 test::FillFn<Scalar>(&in_cpu, [&](int index) { in MakeReductionTestCase()
169 test::FillFn<Scalar>(&test_case->expected, in MakeGatherTestCase()
179 test::FillFn<Scalar>(&in_cpu, [&](int index) { in MakeGatherTestCase()
215 test::FillFn<Scalar>(&test_case->expected, in MakeBroadcastTestCase()
230 test::FillFn<Scalar>(&in_cpu, in MakeBroadcastTestCase()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | memmapped_file_system_test.cc | 42 test::FillFn<float>(test_tensor, in CreateMemmappedFileSystemFile()
54 test::FillFn<float>(test_tensor, in CreateMemmappedFileSystemFile()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | mirror_pad_op_test.cc | 107 test::FillFn<float>(&expected, [=](int i) -> float { in REGISTER_TEST()
141 test::FillFn<float>(&expected, [=](int i) -> float { in TEST_F()
|
| D | crop_and_resize_op_test.cc | 439 test::FillFn<float>(&result1, [=](int i) -> float { return i % kHalf; }); in TEST_F()
446 test::FillFn<float>(&result2, in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_test.cc | 59 test::FillFn<int32>(&csr_row_ptr, [](int unused) { return 0; }); in TEST()
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model_test.cc | 40 test::FillFn<float>(&constant_tensor, [](int) -> float { return 3.0; }); in CreateTestGraph()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_testutil.h | 91 void FillFn(Tensor* tensor, std::function<T(int)> fn) { in FillFn() function
|
| D | tensor_testutil_test.cc | 194 test::FillFn<float>(&x, [](int i) { return 1.0f; }); in TEST()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 473 test::FillFn<float>(&tensor, [&](int i) -> float { in RandomTensor()
486 test::FillFn<double>(&tensor, [&](int i) -> double { in RandomTensor()
499 test::FillFn<complex64>(&tensor, [&](int i) { in RandomTensor()
516 test::FillFn<int32>(&tensor, [&](int i) -> int32 { in RandomTensor()
530 test::FillFn<int64>(&tensor, [&](int i) -> int64 { in RandomTensor()
543 test::FillFn<bool>(&tensor, [&](int i) -> bool { in RandomTensor()
569 test::FillFn<float>(&tensor, [this, &distribution](int i) -> float { in RandomNonNegativeTensor()
576 test::FillFn<double>(&tensor, [this, &distribution](int i) -> double { in RandomNonNegativeTensor()
583 test::FillFn<int32>(&tensor, [this, &distribution](int i) -> int32 { in RandomNonNegativeTensor()
590 test::FillFn<int64>(&tensor, [this, &distribution](int i) -> int64 { in RandomNonNegativeTensor()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 941 FunctionDef FillFn() { in FillFn() function
963 *flib.add_function() = FillFn(); in TEST_F()
965 TF_ASSERT_OK(flib_def_->AddFunctionDef(FillFn())); in TEST_F()
1402 FunctionDef fn = FillFn(); in TEST_F()
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle_test.cc | 583 test::FillFn<float>(&expected_val, [](int offset) -> float { in TEST()
614 test::FillFn<float>(&expected_val, [](int offset) -> float { in TEST()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 675 test::FillFn<float>(&input_tensor, [](int) { return 100.0f; }); in TestQuantizeResizeBilinear()
|