/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 | 77 test::FillFn<qint32>(&quantized_tensor, [](int n) { return qint32(n); }); in BM_RequantizationRange()
|
D | crop_and_resize_op_test.cc | 440 test::FillFn<float>(&result1, [](int i) -> float { return i % kHalf; }); in TEST_F() 447 test::FillFn<float>(&result2, in TEST_F()
|
D | segment_reduction_ops_test.cc | 57 test::FillFn<Index>(&input2, [&num_rows, &segment_size](Index i) -> Index { in BM_SegmentReduction()
|
D | ops_testutil.h | 117 test::FillFn(AddInput(DataTypeToEnum<T>::v(), shape), input_mapping); in AddInput()
|
D | restore_v2_op_test.cc | 43 test::FillFn(&input, input_mapping); in MakeInput()
|
D | restore_op_test.cc | 57 test::FillFn(&input, input_mapping); in MakeInput()
|
/external/tensorflow/tensorflow/contrib/util/ |
D | convert_graphdef_memmapped_format_test.cc | 49 test::FillFn<float>(&test_tensor1, [](int) -> float { return 2.0; }); in TEST() 52 test::FillFn<float>(&test_tensor2, [](int) -> float { return 3.0; }); in TEST() 101 test::FillFn<string>(&test_tensor1, [](int) -> string { return "ABC"; }); in TEST() 104 test::FillFn<string>(&test_tensor2, [](int) -> string { return "XYZ"; }); in TEST()
|
/external/tensorflow/tensorflow/core/nccl/ |
D | nccl_manager_test.cc | 82 test::FillFn<Scalar>(&test_case->expected, in MakeReductionTestCase() 85 test::FillFn<Scalar>(&test_case->expected, in MakeReductionTestCase() 88 test::FillFn<Scalar>(&test_case->expected, [](int) { return -max_; }); in MakeReductionTestCase() 90 test::FillFn<Scalar>(&test_case->expected, [](int) { return max_; }); in MakeReductionTestCase() 102 test::FillFn<Scalar>(&in_cpu, [&](int index) { in MakeReductionTestCase() 141 test::FillFn<Scalar>(&test_case->expected, in MakeGatherTestCase() 151 test::FillFn<Scalar>(&in_cpu, [&](int index) { in MakeGatherTestCase()
|
/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/tools/benchmark/ |
D | benchmark_model_test.cc | 40 test::FillFn<float>(&constant_tensor, [](int) -> float { return 3.0; }); in CreateTestGraph()
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 472 test::FillFn<float>(&tensor, [&](int i) -> float { in RandomTensor() 485 test::FillFn<double>(&tensor, [&](int i) -> double { in RandomTensor() 498 test::FillFn<complex64>(&tensor, [&](int i) { in RandomTensor() 515 test::FillFn<int32>(&tensor, [&](int i) -> int32 { in RandomTensor() 529 test::FillFn<int64>(&tensor, [&](int i) -> int64 { in RandomTensor() 542 test::FillFn<bool>(&tensor, [&](int i) -> bool { in RandomTensor() 568 test::FillFn<float>(&tensor, [this, &distribution](int i) -> float { in RandomNonNegativeTensor() 575 test::FillFn<double>(&tensor, [this, &distribution](int i) -> double { in RandomNonNegativeTensor() 582 test::FillFn<int32>(&tensor, [this, &distribution](int i) -> int32 { in RandomNonNegativeTensor() 589 test::FillFn<int64>(&tensor, [this, &distribution](int i) -> int64 { in RandomNonNegativeTensor() [all …]
|
/external/tensorflow/tensorflow/core/framework/ |
D | tensor_testutil.h | 91 void FillFn(Tensor* tensor, std::function<T(int)> fn) { in FillFn() function
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_compiler_test.cc | 875 FunctionDef FillFn() { in FillFn() function 897 *flib.add_function() = FillFn(); in TEST_F() 899 TF_ASSERT_OK(flib_def_->AddFunctionDef(FillFn())); in TEST_F() 1333 FunctionDef fn = FillFn(); in TEST_F()
|
/external/tensorflow/tensorflow/core/util/tensor_bundle/ |
D | tensor_bundle_test.cc | 345 test::FillFn<float>(&expected_val, [](int offset) -> float { in TEST() 376 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()
|