Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Dmath_utils_test.cc49 Tensor expected_tensor = in TEST() local
51 test::ExpectEqual(quantized_tensor, expected_tensor); in TEST()
70 Tensor expected_tensor = in TEST() local
72 test::ExpectTensorNear<float>(dequantized_tensor, expected_tensor, 1e-6); in TEST()
90 Tensor expected_tensor = in TEST() local
92 test::ExpectEqual(quantized_tensor, expected_tensor); in TEST()
113 Tensor expected_tensor = in TEST() local
115 test::ExpectEqual(quantized_tensor, expected_tensor); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dnode_matchers.cc70 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_t>(tensor, expected_tensor, listener); in MatchAndExplainTensor()
[all …]
/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
44 self.assertEqual(actual_tensor_info.name, expected_tensor.name)
45 self.assertEqual(actual_tensor_info.dtype, expected_tensor.dtype)
47 len(actual_tensor_info.tensor_shape.dim), len(expected_tensor.shape))
50 expected_tensor.shape[i])
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A 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()
/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
141 EXPECT_EQ(output_tensor->Type(), expected_tensor->Type()); in TEST_P()
142 EXPECT_EQ(output_tensor->NumElements(), expected_tensor->NumElements()); in TEST_P()
145 output_tensor->Data(), expected_tensor->Data()); in TEST_P()
/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
47 self.assertAllEqual(expected_tensor.indices, result.indices)
48 self.assertAllEqual(expected_tensor.values, result.values)
50 expected_tensor.dense_shape is not None):
51 self.assertAllEqual(expected_tensor.dense_shape, result.dense_shape)
54 expected_tensor = _indexedslice(expected_array)
55 self._assertEqual_indexedslices(expected_tensor, result)
171 expected_tensor = _indexedslice(tensor_to_add)
172 expected_tensors.append(expected_tensor)
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstring_ngrams_op_test.cc56 Tensor expected_tensor( in assert_string_equal() local
59 test::FillValues<tstring>(&expected_tensor, expected); in assert_string_equal()
60 test::ExpectTensorEqual<tstring>(expected_tensor, value); in assert_string_equal()
64 Tensor expected_tensor( in assert_int64_equal() local
67 test::FillValues<int64_t>(&expected_tensor, expected); in assert_int64_equal()
68 test::ExpectTensorEqual<int64_t>(expected_tensor, value); in assert_int64_equal()
H A Dunary_ops_composition_test.cc49 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()
H A Ddequantize_op_test.cc178 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()
/aosp_15_r20/external/federated-compute/fcp/artifact_building/
H A Dtensor_utils_test.py121 expected_tensor, restored_tensor = sess.run([tensor, restored])
122 self.assertAllEqual(expected_tensor, restored_tensor)
146 expected_tensor, restored_tensor = sess.run([tensor, restored])
147 self.assertAllEqual(expected_tensor, restored_tensor)
/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
97 test::FillValues<T>(&expected_tensor, {expected_scalar_value}); in RunComposedOp()
105 test::ExpectClose(expected_tensor, output_on_host, /*atol=*/1e-5, in RunComposedOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mlir_generated/
H A Dbase_unary_ops_test.h73 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()
H A Dbase_binary_ops_test.h81 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()
/aosp_15_r20/external/pytorch/test/distributed/algorithms/quantization/
H A Dtest_quantization.py291 expected_tensor = torch.cat(
297 expected_tensor = expected_tensor.cuda(rank_to_GPU)
309 self.assertEqual(out_tensor, expected_tensor)
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_multi_threaded_pg.py181 expected_tensor = torch.ones(3, 3) * dist.get_rank() * self.world_size
182 self.assertEqual(output_tensor, expected_tensor)
186 expected_tensor = torch.ones(3, 3) * dist.get_rank()
187 self.assertEqual(output_tensor, expected_tensor)
/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() 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()
213 if (expected_tensor->buffer > 0) { in ExpectSameModels()
[all …]
H A Dquantize_weights_test.cc143 bool ExpectEqualTensor(const Tensor* tensor, const Tensor* expected_tensor) { in ExpectEqualTensor() argument
145 return (tensor->is_variable() == expected_tensor->is_variable()) && in ExpectEqualTensor()
147 GetAsVector(expected_tensor->shape())) && in ExpectEqualTensor()
148 (tensor->name()->str() == expected_tensor->name()->str()); in ExpectEqualTensor()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dget_options_op_test.cc89 Tensor expected_tensor = in TEST_F() local
95 TF_EXPECT_OK(ExpectEqual(expected_tensor, result_tensor)); in TEST_F()
/aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/
H A Dtest_tp_style.py198 expected_tensor = torch.ones(2 * self.world_size, 16, device=self.device_type)
206 self.assertEqual(output, expected_tensor)
336 expected_tensor = torch.ones(8 // self.world_size, 16, device=self.device_type)
344 self.assertEqual(output, expected_tensor)
/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
102 EXPECT_EQ(tensor->buffer, expected_tensor->buffer); in ExpectSameModels()
103 EXPECT_EQ(tensor->is_variable, expected_tensor->is_variable); in ExpectSameModels()
104 EXPECT_EQ(tensor->shape, expected_tensor->shape); in ExpectSameModels()
105 EXPECT_EQ(tensor->name, expected_tensor->name); in ExpectSameModels()
106 EXPECT_EQ(tensor->type, expected_tensor->type); in ExpectSameModels()
109 expected_tensor->quantization.get(); in ExpectSameModels()
1387 const auto expected_tensor = expected_graph->tensors[i].get(); in TEST_F() local
1388 EXPECT_EQ(tensor->buffer, expected_tensor->buffer); in TEST_F()
1389 EXPECT_EQ(tensor->is_variable, expected_tensor->is_variable); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_dynamic_partition_op_test.py172 expected_tensor = ragged_factory_ops.constant(
178 self.assertAllEqual(result, expected_tensor)
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dtest_util.py55 expected_tensor = ops.convert_to_tensor_v2_with_dispatch(
61 assert_op = check_ops.assert_equal(dx, expected_tensor)
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py1618 expected_tensor = _build_tensor(src + 1)
1623 self.assertEqual(output_tensor, expected_tensor)
1690 expected_tensor = _build_tensor(recv_size)
1693 self.assertEqual(output_tensor, expected_tensor)
2024 expected_tensor = _build_tensor(src + 1, value, dtype)
2026 expected_tensor = expected_tensor.cuda(rank_to_GPU[rank][0])
2036 [expected_tensor],
2044 expected_tensor,
2068 self.assertEqual(tensor.size(), expected_tensor.size())
2070 tensor.ne(expected_tensor).max(), torch.tensor(False)
[all …]
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_redistribute.py45 expected_tensor = torch.randn(
48 dtensor = distribute_tensor(expected_tensor, device_mesh, shard_spec)
52 self.assertEqual(expected_tensor, reshard_dtensor.to_local())
/aosp_15_r20/external/tensorflow/tensorflow/python/training/experimental/
H A Dloss_scale_optimizer_test.py90 expected_tensor = ops.convert_to_tensor_v2(
96 assert_op = check_ops.assert_equal(dx, expected_tensor)

12