Home
last modified time | relevance | path

Searched defs:expected_outputs (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dlookup_table_init_op.cc58 DataTypeVector expected_outputs = {}; in Compute() local
126 DataTypeVector expected_outputs = {}; in Compute() local
Dbarrier_ops.cc593 DataTypeVector expected_outputs = {DT_INT64, DT_STRING}; in ComputeAsync() local
Dlookup_table_op.cc836 DataTypeVector expected_outputs = {table->value_dtype()}; in Compute() local
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc577 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorGetNext()
583 TestIterator* iterator, const std::vector<Tensor>& expected_outputs, in CheckIteratorGetNext()
591 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorGetNext()
612 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorSkip()
632 const DatasetParams& params, const std::vector<Tensor>& expected_outputs) { in CheckSplitProviderFullIteration()
647 const std::vector<Tensor>& expected_outputs) { in CheckSplitProviderShardedIteration()
726 const std::vector<Tensor>& expected_outputs, in CheckIteratorSaveAndRestore()
760 const std::vector<Tensor>& expected_outputs, in CheckIteratorSaveAndRestore()
Doptimize_dataset_op_test.cc86 std::vector<Tensor> expected_outputs = in TEST_F() local
Dsparse_tensor_slice_dataset_op_test.cc108 std::vector<std::vector<Tensor>> expected_outputs; member
307 std::vector<std::vector<Tensor>> expected_outputs; member
Dwindow_dataset_op_test.cc237 std::vector<std::vector<Tensor>> expected_outputs; member
405 std::vector<std::vector<Tensor>> expected_outputs; member
Ddataset_test_base.h380 std::vector<Tensor> expected_outputs; member
400 std::vector<Tensor> expected_outputs; member
469 std::vector<Tensor> expected_outputs; member
Dmap_defun_op_test.cc118 std::vector<Tensor> expected_outputs; member
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_multistream_test.cc40 std::vector<int32> expected_outputs(n); in TEST() local
/external/tensorflow/tensorflow/core/data/
Dstandalone_test.cc274 std::vector<int64> expected_outputs; in TEST() member
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc278 std::unordered_set<string> expected_outputs = {"output0:0", "output1:0"}; in TEST_F() local
308 std::unordered_set<string> expected_outputs = {"output0:0", "output1:0"}; in TEST_F() local
450 std::unordered_set<string> expected_outputs = {"output1:0", "output2:0"}; in TEST_F() local
480 std::unordered_set<string> expected_outputs = {"output1:0", "output2:0", in TEST_F() local
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dcache_dataset_serialization_test.py54 def expected_outputs(self): member in CacheDatasetSerializationTest
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_test.py126 expected_outputs, argument
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device.cc112 int expected_outputs = TFE_OpGetOutputLength(op.get(), "outputs", status); in ExecuteWithSpecialOps() local
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc64 const DataTypeSlice expected_outputs, in MatchSignatureHelper()
255 const DataTypeSlice expected_inputs, const DataTypeSlice expected_outputs) { in MatchSignature()
1032 const DataTypeSlice expected_outputs) { in MatchSignature()
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_test.cc880 float expected_outputs[3][4] = {{1.0f, 1.0f, 1.0f, 1.0f}, in TEST_P() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc3116 std::vector<float> expected_outputs(num_inputs); in TEST_P() local
5101 std::vector<std::vector<float>> expected_outputs; in TEST_P() member
5873 std::vector<std::vector<CType>> expected_outputs; in TEST_P() member
6047 std::vector<std::vector<CType>> expected_outputs; in TEST_P() member
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc567 std::vector<string> expected_outputs{"float: [-1,-1,1]", "float: [2,1,1]", in TEST_F() local
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc916 std::vector<Tensor> expected_outputs, test_outputs; in ExpectTfAndXlaOutputsAreClose() local