Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.cc582 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorGetNext()
588 TestIterator* iterator, const std::vector<Tensor>& expected_outputs, in CheckIteratorGetNext()
596 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorGetNext()
617 const std::vector<Tensor>& expected_outputs, bool compare_order) { in CheckIteratorSkip()
637 const DatasetParams& params, const std::vector<Tensor>& expected_outputs) { in CheckSplitProviderFullIteration()
652 const std::vector<Tensor>& expected_outputs) { in CheckSplitProviderShardedIteration()
741 const std::vector<Tensor>& expected_outputs, in CheckIteratorSaveAndRestore()
775 const std::vector<Tensor>& expected_outputs, in CheckIteratorSaveAndRestore()
H A Ddataset_test_base.h426 std::vector<Tensor> expected_outputs; member
446 std::vector<Tensor> expected_outputs; member
515 std::vector<Tensor> expected_outputs; member
H A Dstandalone_test.cc274 std::vector<int64_t> expected_outputs; in TEST() member
/aosp_15_r20/external/tensorflow/tensorflow/cc/tools/
H A Dfreeze_saved_model_test.cc288 std::unordered_set<string> expected_outputs = {"output0:0", "output1:0"}; in TEST_F() local
318 std::unordered_set<string> expected_outputs = {"output0:0", "output1:0"}; in TEST_F() local
464 std::unordered_set<string> expected_outputs = {"output1:0", "output2:0"}; in TEST_F() local
494 std::unordered_set<string> expected_outputs = {"output1:0", "output2:0", in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dlookup_table_init_op.cc61 DataTypeVector expected_outputs = {}; in Compute() local
131 DataTypeVector expected_outputs = {}; in Compute() local
H A Dbarrier_ops.cc589 DataTypeVector expected_outputs = {DT_INT64, DT_STRING}; in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dsparse_tensor_slice_dataset_op_test.cc112 std::vector<std::vector<Tensor>> expected_outputs; member
311 std::vector<std::vector<Tensor>> expected_outputs; member
H A Dwindow_dataset_op_test.cc242 std::vector<std::vector<Tensor>> expected_outputs; member
410 std::vector<std::vector<Tensor>> expected_outputs; member
H A Drewrite_dataset_op_test.cc75 std::vector<Tensor> expected_outputs = in TEST_F() local
H A Doptimize_dataset_op_test.cc86 std::vector<Tensor> expected_outputs = CreateTensors<int64_t>( in TEST_F() local
H A Dmap_defun_op_test.cc118 std::vector<Tensor> expected_outputs; member
/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/simple_hash_table/
H A Dsimple_hash_table_kernel.cc178 DataTypeVector expected_outputs = {DataTypeToEnum<V>::v()}; in Compute() local
236 DataTypeVector expected_outputs = {DataTypeToEnum<K>::v(), in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Drandom_ops_test.cc334 const std::vector<std::vector<int32_t>> expected_outputs = { in TEST() local
359 const std::vector<std::vector<int64_t>> expected_outputs = { in TEST() local
/aosp_15_r20/external/armnn/delegate/python/test/
H A Dutils.py48 def compare_outputs(outputs, expected_outputs): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dgpu_multistream_test.cc41 std::vector<int32_t> expected_outputs(n); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/
H A Dtf_driver.h57 const std::vector<std::pair<string, string>>& expected_outputs, in CheckResults() argument
H A Dtest_runner_test.cc40 const std::vector<std::pair<string, string>>& expected_outputs, in CheckResults() argument
/aosp_15_r20/external/googleapis/google/devtools/remoteworkers/v1test2/
H A Dcommand.proto138 Outputs expected_outputs = 4; field
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dtfrt_op_kernel.h75 const DataTypeSlice expected_outputs) { in MatchSignature()
/aosp_15_r20/external/tensorflow/tensorflow/python/training/experimental/
H A Dloss_scale_test.py122 expected_outputs, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.cc70 const DataTypeSlice expected_outputs, in MatchSignatureHelper()
284 const DataTypeSlice expected_inputs, const DataTypeSlice expected_outputs) { in MatchSignature()
1043 const DataTypeSlice expected_outputs) { in MatchSignature()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device.cc114 int expected_outputs = TFE_OpGetOutputLength(op.get(), "outputs", status); in ExecuteWithSpecialOps() local
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Dcache_test.py441 def expected_outputs(self): member in CacheCheckpointTest
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dlegacy_vmap_test.cpp395 static void checkBroadcastingVmapTransform(TensorList inputs, TensorList expected_outputs) { in checkBroadcastingVmapTransform()
879 static void checkMultiBatchVmapTransform(TensorList inputs, TensorList expected_outputs) { in checkMultiBatchVmapTransform()
/aosp_15_r20/external/executorch/backends/vulkan/test/utils/
H A Dtest_utils.cpp516 std::vector<float> expected_outputs) { in execute_graph_and_check_output()

12