Home
last modified time | relevance | path

Searched refs:src_tensor (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dpadding_test_util.cc29 TensorFloat32 src_tensor; in PaddingAppendWidthTest() local
30 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingAppendWidthTest()
31 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in PaddingAppendWidthTest()
48 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PaddingAppendWidthTest()
59 TensorFloat32 src_tensor; in PaddingPrependWidthTest() local
60 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingPrependWidthTest()
61 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in PaddingPrependWidthTest()
78 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PaddingPrependWidthTest()
89 TensorFloat32 src_tensor; in PaddingAppendHeightTest() local
90 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingAppendHeightTest()
[all …]
Dsoftmax_test_util.cc30 TensorFloat32 src_tensor; in SoftmaxTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 2); in SoftmaxTest()
32 src_tensor.data = {std::log(1.0f), std::log(2.0f), std::log(3.0f), in SoftmaxTest()
46 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in SoftmaxTest()
57 TensorFloat32 src_tensor; in SoftmaxBigNumberTest() local
58 src_tensor.shape = BHWC(1, 2, 1, 2); in SoftmaxBigNumberTest()
61 src_tensor.data.resize(4); in SoftmaxBigNumberTest()
62 src_tensor.data[0] = doubles[0]; in SoftmaxBigNumberTest()
63 src_tensor.data[1] = doubles[1]; in SoftmaxBigNumberTest()
64 src_tensor.data[2] = doubles[2]; in SoftmaxBigNumberTest()
[all …]
Dreduce_test_util.cc29 TensorFloat32 src_tensor; in MeanHWTest() local
30 src_tensor.shape = BHWC(1, 2, 2, 1); in MeanHWTest()
31 src_tensor.data = {1.0f, 2.0f, 3.0f, 4.0f}; in MeanHWTest()
44 CreateReduce(axis, src_tensor.shape, OperationType::MEAN, op_def, in MeanHWTest()
47 src_tensor, absl::make_unique<Reduce>(std::move(operation)), in MeanHWTest()
56 TensorFloat32 src_tensor; in ReduceSumChannelsTest() local
57 src_tensor.shape = BHWC(1, 2, 1, 5); in ReduceSumChannelsTest()
58 src_tensor.data = {1.1, 2.1, 0.7, 0.3, 1.2, 3.1, 4.1, 0.0, 1.0, 4.4}; in ReduceSumChannelsTest()
71 CreateReduce(axis, src_tensor.shape, OperationType::REDUCE_SUM, in ReduceSumChannelsTest()
74 src_tensor, absl::make_unique<Reduce>(std::move(operation)), in ReduceSumChannelsTest()
[all …]
Dresize_test_util.cc29 TensorFloat32 src_tensor; in ResizeBilinearAlignedTest() local
30 src_tensor.shape = BHWC(1, 2, 3, 1); in ResizeBilinearAlignedTest()
31 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f}; in ResizeBilinearAlignedTest()
49 src_tensor, absl::make_unique<Resize>(std::move(operation)), in ResizeBilinearAlignedTest()
61 TensorFloat32 src_tensor; in ResizeBilinearNonAlignedTest() local
62 src_tensor.shape = BHWC(1, 2, 3, 1); in ResizeBilinearNonAlignedTest()
63 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f}; in ResizeBilinearNonAlignedTest()
81 src_tensor, absl::make_unique<Resize>(std::move(operation)), in ResizeBilinearNonAlignedTest()
93 TensorFloat32 src_tensor; in ResizeBilinearWithoutHalfPixelTest() local
94 src_tensor.shape = BHWC(1, 2, 2, 1); in ResizeBilinearWithoutHalfPixelTest()
[all …]
Drelu_test_util.cc29 TensorFloat32 src_tensor; in ReLUNoClipNoAlphaTest() local
30 src_tensor.shape = BHWC(1, 2, 1, 2); in ReLUNoClipNoAlphaTest()
31 src_tensor.data = {-0.5f, 0.8f, -0.6f, 3.2f}; in ReLUNoClipNoAlphaTest()
48 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in ReLUNoClipNoAlphaTest()
58 TensorFloat32 src_tensor; in ReLUClipTest() local
59 src_tensor.shape = BHWC(1, 2, 1, 2); in ReLUClipTest()
60 src_tensor.data = {-0.5f, 0.8f, -0.6f, 3.2f}; in ReLUClipTest()
77 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in ReLUClipTest()
87 TensorFloat32 src_tensor; in ReLUAlphaTest() local
88 src_tensor.shape = BHWC(1, 2, 1, 2); in ReLUAlphaTest()
[all …]
Dwinograd_test_util.cc30 TensorFloat32 src_tensor; in Winograd4x4To36TileX6Test() local
31 src_tensor.shape = BHWC(1, 4, 4, 1); in Winograd4x4To36TileX6Test()
32 src_tensor.data.resize(16); in Winograd4x4To36TileX6Test()
34 src_tensor.data[i] = sin(i); in Winograd4x4To36TileX6Test()
50 const int index = src_tensor.shape.LinearIndex({0, y - 1, i - 1, 0}); in Winograd4x4To36TileX6Test()
51 sum += src_tensor.data[index] * b_t[x * 6 + i]; in Winograd4x4To36TileX6Test()
86 src_tensor, in Winograd4x4To36TileX6Test()
96 TensorFloat32 src_tensor; in Winograd36To4x4Tile4x1Test() local
97 src_tensor.shape = BHWC(1, 36, 1, 1); in Winograd36To4x4Tile4x1Test()
98 src_tensor.data.resize(36); in Winograd36To4x4Tile4x1Test()
[all …]
Dquantize_and_dequantize_test_util.cc28 TensorFloat32 src_tensor; in QuantAndDequant_Dim2Bits8Test() local
29 src_tensor.shape = BHWC(1, 3, 2, 1); in QuantAndDequant_Dim2Bits8Test()
30 src_tensor.data = {0.0f, 1.0f, 0.25f, 0.50f, 0.4444444f, 0.00001f}; in QuantAndDequant_Dim2Bits8Test()
53 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in QuantAndDequant_Dim2Bits8Test()
65 TensorFloat32 src_tensor; in QuantAndDequant_Dim3Bits8_NegativeRangeTest() local
66 src_tensor.shape = BHWC(1, 3, 1, 2); in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
67 src_tensor.data = {0.0f, -0.9f, 0.25f, 0.50f, 0.4444444f, -0.00001f}; in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
90 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
101 TensorFloat32 src_tensor; in QuantAndDequant_Dim3Bits16Test() local
102 src_tensor.shape = BHWC(1, 3, 1, 2); in QuantAndDequant_Dim3Bits16Test()
[all …]
Dspace_to_depth_test_util.cc30 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x1BlockSize2Test() local
31 src_tensor.shape = BHWC(1, 2, 2, 1); in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
32 src_tensor.data = {half(1.0f), half(2.0f), half(3.0f), half(4.0f)}; in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
44 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
56 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x2BlockSize2Test() local
57 src_tensor.shape = BHWC(1, 2, 2, 2); in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
58 src_tensor.data = {half(1.4f), half(2.3f), half(3.2f), half(4.1f), in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
71 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
84 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x3BlockSize2Test() local
85 src_tensor.shape = BHWC(1, 2, 2, 3); in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
[all …]
Dmean_stddev_normalization_test_util.cc33 TensorFloat32 src_tensor; in MeanStddevNormSeparateBatchesTest() local
34 src_tensor.shape = BHWC(1, 1, 1, 4); in MeanStddevNormSeparateBatchesTest()
35 src_tensor.data = {mean - 2 * diff, mean - diff, mean + diff, in MeanStddevNormSeparateBatchesTest()
48 {src_tensor}, in MeanStddevNormSeparateBatchesTest()
69 TensorFloat32 src_tensor; in MeanStddevNormalizationAllBatchesTest() local
70 src_tensor.shape = BHWC(9, 1, 1, 4); in MeanStddevNormalizationAllBatchesTest()
71 src_tensor.data = { in MeanStddevNormalizationAllBatchesTest()
95 {src_tensor}, in MeanStddevNormalizationAllBatchesTest()
126 TensorFloat32 src_tensor; in MeanStddevNormalizationLargeVectorTest() local
127 src_tensor.shape = BHWC(1, 1, 1, kVectorSize); in MeanStddevNormalizationLargeVectorTest()
[all …]
Dpooling_test_util.cc29 TensorFloat32 src_tensor; in AveragePoolingTest() local
30 src_tensor.shape = BHWC(1, 2, 2, 2); in AveragePoolingTest()
31 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in AveragePoolingTest()
51 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in AveragePoolingTest()
60 TensorFloat32 src_tensor; in AveragePoolingNonEmptyPaddingTest() local
61 src_tensor.shape = BHWC(1, 2, 2, 1); in AveragePoolingNonEmptyPaddingTest()
62 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in AveragePoolingNonEmptyPaddingTest()
82 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in AveragePoolingNonEmptyPaddingTest()
92 TensorFloat32 src_tensor; in MaxPoolingTest() local
93 src_tensor.shape = BHWC(1, 2, 2, 2); in MaxPoolingTest()
[all …]
Dconv_powervr_test_util.cc29 TensorFloat32 src_tensor; in ConvPowerVR1x1SimpleWeightsTest() local
30 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvPowerVR1x1SimpleWeightsTest()
31 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvPowerVR1x1SimpleWeightsTest()
55 src_tensor, absl::make_unique<ConvPowerVR>(std::move(operation)), in ConvPowerVR1x1SimpleWeightsTest()
66 TensorFloat32 src_tensor; in ConvPowerVR1x1Test() local
67 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvPowerVR1x1Test()
68 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvPowerVR1x1Test()
92 src_tensor, absl::make_unique<ConvPowerVR>(std::move(operation)), in ConvPowerVR1x1Test()
103 TensorFloat32 src_tensor; in ConvPowerVRSimpleWeightsTest() local
104 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvPowerVRSimpleWeightsTest()
[all …]
Dprelu_test_util.cc29 TensorFloat32 src_tensor; in PReLUAlphaTest() local
30 src_tensor.shape = BHWC(1, 2, 1, 2); in PReLUAlphaTest()
31 src_tensor.data = {0.0f, -1.0f, -2.0f, 3.0f}; in PReLUAlphaTest()
51 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PReLUAlphaTest()
61 TensorFloat32 src_tensor; in PReLUAlphaClipTest() local
62 src_tensor.shape = BHWC(1, 2, 1, 2); in PReLUAlphaClipTest()
63 src_tensor.data = {0.0f, -1.0f, -2.0f, 3.0f}; in PReLUAlphaClipTest()
83 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PReLUAlphaClipTest()
93 TensorFloat32 src_tensor; in PReLUHWCAlphaTest() local
94 src_tensor.shape = BHWC(1, 2, 1, 2); in PReLUHWCAlphaTest()
[all …]
Ddepthwise_conv_test_util.cc29 TensorFloat32 src_tensor; in DepthwiseConvSimpleWeightsTest() local
30 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConvSimpleWeightsTest()
31 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in DepthwiseConvSimpleWeightsTest()
55 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in DepthwiseConvSimpleWeightsTest()
66 TensorFloat32 src_tensor; in DepthwiseConvNoMultiplierTest() local
67 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConvNoMultiplierTest()
68 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in DepthwiseConvNoMultiplierTest()
92 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in DepthwiseConvNoMultiplierTest()
103 TensorFloat32 src_tensor; in DepthwiseConvMultiplier2Test() local
104 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConvMultiplier2Test()
[all …]
Dfully_connected_test_util.cc29 TensorFloat32 src_tensor; in FullyConnectedTest() local
30 src_tensor.shape = BHWC(1, 1, 1, 4); in FullyConnectedTest()
31 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in FullyConnectedTest()
52 src_tensor, absl::make_unique<FullyConnected>(std::move(operation)), in FullyConnectedTest()
62 TensorFloat32 src_tensor; in FullyConnectedLargeTest() local
63 src_tensor.shape = BHWC(1, 1, 1, 8); in FullyConnectedLargeTest()
64 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in FullyConnectedLargeTest()
98 src_tensor, absl::make_unique<FullyConnected>(std::move(operation)), in FullyConnectedLargeTest()
113 TensorFloat32 src_tensor; in FullyConnectedExtraLargeTest() local
114 src_tensor.shape = BHWC(1, 1, 1, kInputSize); in FullyConnectedExtraLargeTest()
[all …]
Dreshape_test_util.cc30 TensorFloat32 src_tensor; in ReshapeTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 3); in ReshapeTest()
32 src_tensor.data = {half(0.5f), half(-1.1f), half(-2.2f), in ReshapeTest()
45 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in ReshapeTest()
56 TensorFloat32 src_tensor; in Reshapex4Test() local
57 src_tensor.shape = BHWC(1, 1, 1, 8); in Reshapex4Test()
58 src_tensor.data = {half(0.5f), half(-1.1f), half(-2.2f), half(3.1f), in Reshapex4Test()
71 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in Reshapex4Test()
Delementwise_test_util.cc29 TensorFloat32 src_tensor; in AbsTest() local
30 src_tensor.shape = BHWC(1, 2, 1, 2); in AbsTest()
31 src_tensor.data = {half(0.0f), half(-1.0f), half(-0.05f), half(0.045f)}; in AbsTest()
44 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in AbsTest()
55 TensorFloat32 src_tensor; in CosTest() local
56 src_tensor.shape = BHWC(1, 2, 1, 2); in CosTest()
57 src_tensor.data = {0.0f, -1.0f, -0.05f, 0.045f}; in CosTest()
71 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in CosTest()
82 TensorFloat32 src_tensor; in CopyTest() local
83 src_tensor.shape = BHWC(1, 2, 1, 2); in CopyTest()
[all …]
Dconvolution_transposed_test_util.cc30 TensorFloat32 src_tensor; in ConvolutionTransposedSimpleWeightsTest() local
31 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvolutionTransposedSimpleWeightsTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvolutionTransposedSimpleWeightsTest()
56 src_tensor, in ConvolutionTransposedSimpleWeightsTest()
71 TensorFloat32 src_tensor; in ConvolutionTransposedTest() local
72 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvolutionTransposedTest()
73 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvolutionTransposedTest()
96 src_tensor, in ConvolutionTransposedTest()
Ddepthwise_conv_3x3_test_util.cc29 TensorFloat32 src_tensor; in DepthwiseConv3x3SimpleWeightsTest() local
30 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConv3x3SimpleWeightsTest()
31 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in DepthwiseConv3x3SimpleWeightsTest()
56 src_tensor, absl::make_unique<DepthwiseConv3x3>(std::move(operation)), in DepthwiseConv3x3SimpleWeightsTest()
67 TensorFloat32 src_tensor; in DepthwiseConv3x3Test() local
68 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConv3x3Test()
69 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in DepthwiseConv3x3Test()
94 src_tensor, absl::make_unique<DepthwiseConv3x3>(std::move(operation)), in DepthwiseConv3x3Test()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconv_buffer_1x1_test.cc35 TensorFloat32 src_tensor; in TEST_F() local
36 src_tensor.shape = BHWC(1, 2, 1, 4); in TEST_F()
37 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in TEST_F()
60 creation_context_.GetGpuInfo(), op_def, attr, &src_tensor.shape); in TEST_F()
62 src_tensor, creation_context_, in TEST_F()
71 TensorFloat32 src_tensor; in TEST_F() local
72 src_tensor.shape = BHWC(1, 2, 1, 4); in TEST_F()
73 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in TEST_F()
97 creation_context_.GetGpuInfo(), op_def, attr, &src_tensor.shape); in TEST_F()
99 src_tensor, creation_context_, in TEST_F()
Dconvolution_transposed_3x3_thin_test.cc35 TensorFloat32 src_tensor; in TEST_F() local
36 src_tensor.shape = BHWC(1, 2, 2, 1); in TEST_F()
37 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in TEST_F()
61 src_tensor, creation_context_, in TEST_F()
73 TensorFloat32 src_tensor; in TEST_F() local
74 src_tensor.shape = BHWC(1, 2, 2, 1); in TEST_F()
75 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in TEST_F()
99 src_tensor, creation_context_, in TEST_F()
Dconvolution_transposed_thin_test.cc35 TensorFloat32 src_tensor; in TEST_F() local
36 src_tensor.shape = BHWC(1, 2, 2, 2); in TEST_F()
37 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in TEST_F()
61 src_tensor, creation_context_, in TEST_F()
76 TensorFloat32 src_tensor; in TEST_F() local
77 src_tensor.shape = BHWC(1, 2, 2, 2); in TEST_F()
78 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in TEST_F()
101 src_tensor, creation_context_, in TEST_F()
Dconv_constants_test.cc35 TensorFloat32 src_tensor; in TEST_F() local
36 src_tensor.shape = BHWC(1, 2, 2, 2); in TEST_F()
37 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in TEST_F()
61 src_tensor, creation_context_, in TEST_F()
71 TensorFloat32 src_tensor; in TEST_F() local
72 src_tensor.shape = BHWC(1, 2, 2, 2); in TEST_F()
73 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in TEST_F()
98 src_tensor, creation_context_, in TEST_F()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_aggregate_ops.cc43 const Tensor& src_tensor = MklGetInput(ctx, src_index); in GetTensorShape() local
47 : src_tensor.shape(); in GetTensorShape()
96 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in ComputeScalar() local
97 T* src_i = const_cast<T*>(src_tensor.flat<T>().data()); in ComputeScalar()
138 const Tensor& src_tensor = MklGetInput(ctx, kSrc0Idx); in Compute() local
143 if (src_tensor.shape().num_elements() == 0) { in Compute()
145 output_tf_shape = src_tensor.shape(); in Compute()
151 if (src_tensor.dims() == 0) { in Compute()
188 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in Compute() local
195 if (src_tensor.dims() == 4) { in Compute()
[all …]
Dmkl_relu_op.cc443 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute() local
446 if (src_tensor.dims() == 0) { in Compute()
454 if (src_tensor.shape().num_elements() == 0) { in Compute()
470 src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute()
485 const T* src_data = src_tensor.flat<T>().data(); in Compute()
487 src.SetUsrMem(src_md, &src_tensor); in Compute()
508 dnn_shape_dst.SetTfLayout(src_tensor.dims(), in Compute()
509 TFShapeToMklDnnDims(src_tensor.shape()), in Compute()
517 tf_shape_dst = src_tensor.shape(); in Compute()
594 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dconv_test.mm44 TensorFloat32 src_tensor;
45 src_tensor.shape = BHWC(1, 2, 2, 1);
46 src_tensor.data = {1, 1, 1, 1};
70 src_tensor, absl::make_unique<ConvolutionMetal>(std::move(operation)), BHWC(1, 2, 2, 2),
80 TensorFloat32 src_tensor;
81 src_tensor.shape = BHWC(1, 3, 3, 1);
82 src_tensor.data = {1, 1, 1, 1, 1, 1, 1, 1, 1};
106 src_tensor, absl::make_unique<ConvolutionMetal>(std::move(operation)), BHWC(1, 1, 1, 1),
116 TensorFloat32 src_tensor;
117 src_tensor.shape = BHWC(1, 2, 2, 1);
[all …]

12