Home
last modified time | relevance | path

Searched refs:offset_tensor (Results 1 – 5 of 5) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_assign.cpp259 Tensor<int, 1> offset_tensor(10); in test_compound_assign() local
261 offset_tensor.setRandom(); in test_compound_assign()
264 tensor += offset_tensor; in test_compound_assign()
266 VERIFY_IS_EQUAL(tensor(i), start_tensor(i) + offset_tensor(i)); in test_compound_assign()
270 tensor -= offset_tensor; in test_compound_assign()
272 VERIFY_IS_EQUAL(tensor(i), start_tensor(i) - offset_tensor(i)); in test_compound_assign()
276 tensor *= offset_tensor; in test_compound_assign()
278 VERIFY_IS_EQUAL(tensor(i), start_tensor(i) * offset_tensor(i)); in test_compound_assign()
282 tensor /= offset_tensor; in test_compound_assign()
284 VERIFY_IS_EQUAL(tensor(i), start_tensor(i) / offset_tensor(i)); in test_compound_assign()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc353 Tensor offset_tensor(DT_FLOAT, TensorShape({6})); in TestQuantizeBiasAdd() local
354 test::FillIota<float>(&offset_tensor, 1); in TestQuantizeBiasAdd()
356 Const(root.WithOpName("offset_op"), Input::Initializer(offset_tensor)); in TestQuantizeBiasAdd()
575 Tensor offset_tensor(DT_QUINT8, TensorShape({6})); in TestRemoveRedundantQuantizationWithBiasAdd() local
576 test::FillValues<quint8>(&offset_tensor, {1, 2, 3, 4, 5, 6}); in TestRemoveRedundantQuantizationWithBiasAdd()
578 Const(root.WithOpName("offset_op"), Input::Initializer(offset_tensor)); in TestRemoveRedundantQuantizationWithBiasAdd()
701 Tensor offset_tensor(DT_QUINT8, TensorShape({6})); in TestRemoveRedundantQuantizationWithMultipleOutputs() local
702 test::FillValues<quint8>(&offset_tensor, {1, 2, 3, 4, 5, 6}); in TestRemoveRedundantQuantizationWithMultipleOutputs()
704 Const(root.WithOpName("offset_op"), Input::Initializer(offset_tensor)); in TestRemoveRedundantQuantizationWithMultipleOutputs()
884 Tensor offset_tensor(DT_FLOAT, TensorShape({6})); in TestConvertFakeQuantsToRequantize() local
[all …]
Dfreeze_requantization_ranges_test.cc65 Tensor offset_tensor(DT_QUINT8, TensorShape({6})); in TestFreezeRequantizationRanges() local
66 test::FillValues<quint8>(&offset_tensor, {1, 2, 3, 4, 5, 6}); in TestFreezeRequantizationRanges()
68 Const(root.WithOpName("offset_op"), Input::Initializer(offset_tensor)); in TestFreezeRequantizationRanges()
/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
Dlayer_norm.cc298 TfLiteTensor offset_tensor = in Eval() local
304 offset = PodDequantize(offset_tensor, 0); in Eval()
307 offset = offset_tensor.data.f[0]; in Eval()
/external/tensorflow/tensorflow/core/kernels/
Dquantized_batch_norm_op.cc120 Tensor offset_tensor(DataTypeToEnum<T2>::v(), {depth}); in FixedPointBatchNorm() local
121 auto offset_flat = offset_tensor.flat<T2>(); in FixedPointBatchNorm()