Home
last modified time | relevance | path

Searched defs:expected_tensor (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Dmath_utils_test.cc49 Tensor expected_tensor = in TEST() local
70 Tensor expected_tensor = in TEST() local
90 Tensor expected_tensor = in TEST() local
113 Tensor expected_tensor = in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstring_ngrams_op_test.cc56 Tensor expected_tensor( in assert_string_equal() local
64 Tensor expected_tensor( in assert_int64_equal() local
H A Dunary_ops_composition_test.cc49 Tensor expected_tensor(allocator(), DataTypeToEnum<T>::value, shape); in RunComposedOp() local
H A Ddequantize_op_test.cc178 Tensor expected_tensor(allocator(), DT_FLOAT, TensorShape(dims)); in RunDequantizeScaledTest() local
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dsimple_save_test.py43 def _check_tensor_info(self, actual_tensor_info, expected_tensor): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dget_options_op_test.cc89 Tensor expected_tensor = in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dunary_ops_composition_test.cc96 Tensor expected_tensor(cpu_allocator, dtype, shape); in RunComposedOp() local
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/
H A Dsaved_variable_loading_test.cc128 AbstractTensorPtr expected_tensor(expected_handle->Resolve(&status)); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dquantize_weights_test.cc138 auto expected_tensor = [](float value) { in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_conditional_accumulator_test.py46 def _assertEqual_indexedslices(self, expected_tensor, result): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dnode_matchers.cc70 bool MatchAndExplainTensor(const Tensor& tensor, const Tensor& expected_tensor, in MatchAndExplainTensor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_model_test.cc157 void ExpectEqualTensor(TensorT* tensor, TensorT* expected_tensor) { in ExpectEqualTensor()
207 auto* expected_tensor = FindMatchingExpectedTensor( in ExpectSameModels() local
H A Dquantize_weights_test.cc143 bool ExpectEqualTensor(const Tensor* tensor, const Tensor* expected_tensor) { in ExpectEqualTensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_model_test.cc101 const auto expected_tensor = expected_graph->tensors[i].get(); in ExpectSameModels() local
1387 const auto expected_tensor = expected_graph->tensors[i].get(); in TEST_F() local
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py6459 self, tensor, expected_tensor, op, reduction_fn=dist.all_reduce, dst=None argument