| /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()
|