Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dlookup_table_init_op.cc61 DataTypeVector expected_outputs = {}; in Compute() local
131 DataTypeVector expected_outputs = {}; in Compute() local
Dbarrier_ops.cc589 DataTypeVector expected_outputs = {DT_INT64, DT_STRING}; in ComputeAsync() local
Dlookup_table_op.cc916 DataTypeVector expected_outputs = {table->value_dtype()}; in Compute() local
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.cc572 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorGetNext()
578 TestIterator* iterator, const std::vector<Tensor>& expected_outputs, in CheckIteratorGetNext()
586 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorGetNext()
607 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorSkip()
627 const DatasetParams& params, const std::vector<Tensor>& expected_outputs) { in CheckSplitProviderFullIteration()
642 const std::vector<Tensor>& expected_outputs) { in CheckSplitProviderShardedIteration()
731 const std::vector<Tensor>& expected_outputs, in CheckIteratorSaveAndRestore()
765 const std::vector<Tensor>& expected_outputs, in CheckIteratorSaveAndRestore()
Dstandalone_test.cc274 std::vector<int64> expected_outputs; in TEST() member
Ddataset_test_base.h416 std::vector<Tensor> expected_outputs; member
436 std::vector<Tensor> expected_outputs; member
505 std::vector<Tensor> expected_outputs; member
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_multistream_test.cc41 std::vector<int32> expected_outputs(n); in TEST() local
/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/core/kernels/data/
Dsparse_tensor_slice_dataset_op_test.cc112 std::vector<std::vector<Tensor>> expected_outputs; member
311 std::vector<std::vector<Tensor>> expected_outputs; member
Doptimize_dataset_op_test.cc86 std::vector<Tensor> expected_outputs = in TEST_F() local
Dwindow_dataset_op_test.cc241 std::vector<std::vector<Tensor>> expected_outputs; member
409 std::vector<std::vector<Tensor>> expected_outputs; member
Dmap_defun_op_test.cc118 std::vector<Tensor> expected_outputs; member
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dtfrt_op_kernel.h75 const DataTypeSlice expected_outputs) { in MatchSignature()
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_test.py126 expected_outputs, argument
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcache_test.py458 def expected_outputs(self): member in CacheCheckpointTest
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device.cc114 int expected_outputs = TFE_OpGetOutputLength(op.get(), "outputs", status); in ExecuteWithSpecialOps() local
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc65 const DataTypeSlice expected_outputs, in MatchSignatureHelper()
248 const DataTypeSlice expected_inputs, const DataTypeSlice expected_outputs) { in MatchSignature()
962 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.cc3370 std::vector<float> expected_outputs(num_inputs); in TEST_P() local
5552 std::vector<std::vector<float>> expected_outputs; in TEST_P() member
6321 std::vector<std::vector<CType>> expected_outputs; in TestConvertSplit() member
6489 std::vector<std::vector<float>> expected_outputs; member
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc719 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