| /external/tensorflow/tensorflow/compiler/jit/ |
| D | node_matchers.cc | 70 bool MatchAndExplainTensor(const Tensor& tensor, const Tensor& expected_tensor, in MatchAndExplainTensor() argument
72 if (tensor.dtype() != expected_tensor.dtype()) { in MatchAndExplainTensor()
75 << DataType_Name(expected_tensor.dtype()) in MatchAndExplainTensor()
83 return CompareTensor<Eigen::half>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
85 return CompareTensor<float>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
87 return CompareTensor<double>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
89 return CompareTensor<int8>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
91 return CompareTensor<int16>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
93 return CompareTensor<int32>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
95 return CompareTensor<int64>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
[all …]
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | simple_save_test.py | 47 def _check_tensor_info(self, actual_tensor_info, expected_tensor): argument
48 self.assertEqual(actual_tensor_info.name, expected_tensor.name)
49 self.assertEqual(actual_tensor_info.dtype, expected_tensor.dtype)
51 len(actual_tensor_info.tensor_shape.dim), len(expected_tensor.shape))
54 expected_tensor.shape[i])
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_weights_test.cc | 138 auto expected_tensor = [](float value) { in TEST_F() local
151 expected_tensor(-5.0), existing_tensor("input_op_quantized_min"), 1e-5); in TEST_F()
153 expected_tensor(0.0), existing_tensor("input_op_quantized_max"), 1e-5); in TEST_F()
157 expected_tensor(0.0), existing_tensor("weights_op_quantized_min"), 1e-5); in TEST_F()
159 expected_tensor(4.0), existing_tensor("weights_op_quantized_max"), 1e-5); in TEST_F()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_variable_loading_test.cc | 127 AbstractTensorPtr expected_tensor(expected_handle->Resolve(&status)); in TEST_P() local
140 EXPECT_EQ(output_tensor->Type(), expected_tensor->Type()); in TEST_P()
141 EXPECT_EQ(output_tensor->NumElements(), expected_tensor->NumElements()); in TEST_P()
144 output_tensor->Data(), expected_tensor->Data()); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_composition_test.cc | 96 Tensor expected_tensor(cpu_allocator, dtype, shape); in RunComposedOp() local
97 test::FillValues<T>(&expected_tensor, {expected_scalar_value}); in RunComposedOp()
105 test::ExpectClose(expected_tensor, output_on_host, /*atol=*/1e-5, in RunComposedOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | string_ngrams_op_test.cc | 56 Tensor expected_tensor(allocator(), DT_STRING, in assert_string_equal() local
58 test::FillValues<tstring>(&expected_tensor, expected); in assert_string_equal()
59 test::ExpectTensorEqual<tstring>(expected_tensor, value); in assert_string_equal()
63 Tensor expected_tensor(allocator(), DT_INT64, in assert_int64_equal() local
65 test::FillValues<int64>(&expected_tensor, expected); in assert_int64_equal()
66 test::ExpectTensorEqual<int64>(expected_tensor, value); in assert_int64_equal()
|
| D | unary_ops_composition_test.cc | 49 Tensor expected_tensor(allocator(), DataTypeToEnum<T>::value, shape); in RunComposedOp() local
50 test::FillValues<T>(&expected_tensor, {expected}); in RunComposedOp()
51 test::ExpectClose(expected_tensor, *GetOutput(0)); in RunComposedOp()
|
| D | dequantize_op_test.cc | 178 Tensor expected_tensor(allocator(), DT_FLOAT, TensorShape(dims)); in RunDequantizeScaledTest() local
179 test::FillValues<float>(&expected_tensor, in RunDequantizeScaledTest()
181 test::ExpectClose(expected_tensor, *GetOutput(0)); in RunDequantizeScaledTest()
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_unary_ops_test.h | 73 Tensor expected_tensor(allocator(), DataTypeToEnum<OutT>::value, shape); in RunAndExpectResult()
74 test::FillValues<OutT>(&expected_tensor, expected_output); in RunAndExpectResult()
76 test::ExpectEqual(expected_tensor, *GetOutput(0), in RunAndExpectResult()
80 test::ExpectClose(expected_tensor, *GetOutput(0), kAbsoluteTolerance, in RunAndExpectResult()
|
| D | base_binary_ops_test.h | 81 Tensor expected_tensor(allocator(), DataTypeToEnum<OutT>::value, in RunAndExpectResult()
83 test::FillValues<OutT>(&expected_tensor, expected_output); in RunAndExpectResult()
85 test::ExpectEqual(expected_tensor, *GetOutput(0), in RunAndExpectResult()
89 test::ExpectClose(expected_tensor, *GetOutput(0), config.atol, in RunAndExpectResult()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | get_options_op_test.cc | 89 Tensor expected_tensor = in TEST_F() local
95 TF_EXPECT_OK(ExpectEqual(expected_tensor, result_tensor)); in TEST_F()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_conditional_accumulator_test.py | 49 def _assertEqual_indexedslices(self, expected_tensor, result): argument
50 self.assertAllEqual(expected_tensor.indices, result.indices)
51 self.assertAllEqual(expected_tensor.values, result.values)
53 expected_tensor.dense_shape is not None):
54 self.assertAllEqual(expected_tensor.dense_shape, result.dense_shape)
57 expected_tensor = _indexedslice(expected_array)
58 self._assertEqual_indexedslices(expected_tensor, result)
174 expected_tensor = _indexedslice(tensor_to_add)
175 expected_tensors.append(expected_tensor)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | test_util.py | 55 expected_tensor = ops.convert_to_tensor_v2_with_dispatch(
61 assert_op = check_ops.assert_equal(dx, expected_tensor)
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_dynamic_partition_op_test.py | 176 expected_tensor = ragged_factory_ops.constant(
182 self.assertAllEqual(result, expected_tensor)
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model_test.cc | 93 const auto expected_tensor = expected_graph->tensors[i].get(); in ExpectSameModels() local
94 EXPECT_EQ(tensor->buffer, expected_tensor->buffer); in ExpectSameModels()
95 EXPECT_EQ(tensor->is_variable, expected_tensor->is_variable); in ExpectSameModels()
96 EXPECT_EQ(tensor->shape, expected_tensor->shape); in ExpectSameModels()
97 EXPECT_EQ(tensor->name, expected_tensor->name); in ExpectSameModels()
98 EXPECT_EQ(tensor->type, expected_tensor->type); in ExpectSameModels()
101 expected_tensor->quantization.get(); in ExpectSameModels()
1367 const auto expected_tensor = expected_graph->tensors[i].get(); in TEST_F() local
1368 EXPECT_EQ(tensor->buffer, expected_tensor->buffer); in TEST_F()
1369 EXPECT_EQ(tensor->is_variable, expected_tensor->is_variable); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/ |
| D | quantize_model_test.cc | 157 void ExpectEqualTensor(TensorT* tensor, TensorT* expected_tensor) { in ExpectEqualTensor() argument
159 EXPECT_NE(expected_tensor, nullptr); in ExpectEqualTensor()
160 EXPECT_EQ(tensor->is_variable, expected_tensor->is_variable); in ExpectEqualTensor()
161 EXPECT_EQ(tensor->shape, expected_tensor->shape); in ExpectEqualTensor()
162 EXPECT_EQ(tensor->type, expected_tensor->type); in ExpectEqualTensor()
164 const auto expected_quantization_params = expected_tensor->quantization.get(); in ExpectEqualTensor()
207 auto* expected_tensor = FindMatchingExpectedTensor( in ExpectSameModels() local
209 if (!expected_tensor) { in ExpectSameModels()
212 ExpectEqualTensor(tensor.get(), expected_tensor); in ExpectSameModels()
215 const int expected_buffer_idx = expected_tensor->buffer; in ExpectSameModels()
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer_test.py | 95 expected_tensor = ops.convert_to_tensor_v2(
101 assert_op = check_ops.assert_equal(dx, expected_tensor)
|