Home
last modified time | relevance | path

Searched refs:expected_tensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dnode_matchers.cc68 bool MatchAndExplainTensor(const Tensor& tensor, const Tensor& expected_tensor, in MatchAndExplainTensor() argument
70 if (tensor.dtype() != expected_tensor.dtype()) { in MatchAndExplainTensor()
73 << DataType_Name(expected_tensor.dtype()) in MatchAndExplainTensor()
81 return CompareTensor<float>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
83 return CompareTensor<double>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
85 return CompareTensor<int8>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
87 return CompareTensor<int16>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
89 return CompareTensor<int32>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
91 return CompareTensor<int64>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
93 return CompareTensor<uint8>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
[all …]
/external/tensorflow/tensorflow/python/saved_model/
Dsimple_save_test.py48 def _check_tensor_info(self, actual_tensor_info, expected_tensor): argument
49 self.assertEqual(actual_tensor_info.name, expected_tensor.name)
50 self.assertEqual(actual_tensor_info.dtype, expected_tensor.dtype)
52 len(actual_tensor_info.tensor_shape.dim), len(expected_tensor.shape))
55 expected_tensor.shape[i])
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_weights_test.cc138 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/python/keras/mixed_precision/experimental/
Dtest_util.py51 expected_tensor = ops.convert_to_tensor(expected_gradient, dtype=dx.dtype)
52 assert_op = check_ops.assert_equal(dx, expected_tensor, data=[dx])
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dtensor_util.py215 def with_same_shape(expected_tensor, tensor): argument
224 with ops.name_scope('%s/' % tensor.op.name, values=[expected_tensor, tensor]):
225 tensor_shape = expected_tensor.get_shape()
228 else array_ops.shape(expected_tensor, name='expected_shape'))
/external/tensorflow/tensorflow/core/kernels/
Dunary_ops_composition_test.cc48 Tensor expected_tensor(allocator(), DataTypeToEnum<T>::value, shape); in RunComposedOp() local
49 test::FillValues<T>(&expected_tensor, {expected}); in RunComposedOp()
50 test::ExpectClose(expected_tensor, *GetOutput(0)); in RunComposedOp()
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_conditional_accumulator_test.py49 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)