| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | softmax_test_util.cc | 31 TensorFloat32 src_tensor; in SoftmaxTest() local
32 src_tensor.shape = BHWC(1, 2, 1, 2); in SoftmaxTest()
33 src_tensor.data = {std::log(1.0f), std::log(2.0f), std::log(3.0f), in SoftmaxTest()
47 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in SoftmaxTest()
58 TensorFloat32 src_tensor; in SoftmaxBigNumberTest() local
59 src_tensor.shape = BHWC(1, 2, 1, 2); in SoftmaxBigNumberTest()
62 src_tensor.data.resize(4); in SoftmaxBigNumberTest()
63 src_tensor.data[0] = doubles[0]; in SoftmaxBigNumberTest()
64 src_tensor.data[1] = doubles[1]; in SoftmaxBigNumberTest()
65 src_tensor.data[2] = doubles[2]; in SoftmaxBigNumberTest()
[all …]
|
| D | mean_stddev_normalization_test_util.cc | 35 TensorFloat32 src_tensor; in MeanStddevNormSeparateBatchesTest() local
36 src_tensor.shape = BHWC(1, 1, 2, 4); in MeanStddevNormSeparateBatchesTest()
37 src_tensor.data = {mean - 2 * diff, mean - diff, mean + diff, in MeanStddevNormSeparateBatchesTest()
49 src_tensor.shape); in MeanStddevNormSeparateBatchesTest()
51 {src_tensor}, in MeanStddevNormSeparateBatchesTest()
70 op_def, env->GetGpuInfo(), src_tensor.shape, in MeanStddevNormSeparateBatchesTest()
73 env->ExecuteGPUOperation({src_tensor}, in MeanStddevNormSeparateBatchesTest()
86 TensorFloat32 src_tensor; in MeanStddevNormalizationAllBatchesTest() local
87 src_tensor.shape = BHWC(9, 1, 1, 4); in MeanStddevNormalizationAllBatchesTest()
88 src_tensor.data = { in MeanStddevNormalizationAllBatchesTest()
[all …]
|
| D | padding_test_util.cc | 30 TensorFloat32 src_tensor; in PaddingAppendWidthTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingAppendWidthTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in PaddingAppendWidthTest()
49 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in PaddingAppendWidthTest()
60 TensorFloat32 src_tensor; in PaddingPrependWidthTest() local
61 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingPrependWidthTest()
62 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in PaddingPrependWidthTest()
79 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in PaddingPrependWidthTest()
90 TensorFloat32 src_tensor; in PaddingAppendHeightTest() local
91 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingAppendHeightTest()
[all …]
|
| D | resize_test_util.cc | 30 TensorFloat32 src_tensor; in ResizeBilinearAlignedTest() local
31 src_tensor.shape = BHWC(1, 2, 3, 1); in ResizeBilinearAlignedTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f}; in ResizeBilinearAlignedTest()
50 src_tensor, std::make_unique<Resize>(std::move(operation)), in ResizeBilinearAlignedTest()
62 TensorFloat32 src_tensor; in ResizeBilinearNonAlignedTest() local
63 src_tensor.shape = BHWC(1, 2, 3, 1); in ResizeBilinearNonAlignedTest()
64 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f}; in ResizeBilinearNonAlignedTest()
82 src_tensor, std::make_unique<Resize>(std::move(operation)), in ResizeBilinearNonAlignedTest()
94 TensorFloat32 src_tensor; in ResizeBilinearWithoutHalfPixelTest() local
95 src_tensor.shape = BHWC(1, 2, 2, 1); in ResizeBilinearWithoutHalfPixelTest()
[all …]
|
| D | tile_test_util.cc | 30 TensorFloat32 src_tensor; in TileChannelsTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 3); in TileChannelsTest()
32 src_tensor.data = {half(1.0f), half(2.0f), half(3.0f), in TileChannelsTest()
42 GPUOperation operation = CreateTile(op_def, src_tensor.shape.c); in TileChannelsTest()
44 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in TileChannelsTest()
57 TensorFloat32 src_tensor; in TileChannelsX4Test() local
58 src_tensor.shape = BHWC(1, 2, 1, 4); in TileChannelsX4Test()
59 src_tensor.data = {half(1.0f), half(2.0f), half(3.0f), half(7.0f), in TileChannelsX4Test()
69 GPUOperation operation = CreateTile(op_def, src_tensor.shape.c); in TileChannelsX4Test()
71 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in TileChannelsX4Test()
[all …]
|
| D | winograd_test_util.cc | 31 TensorFloat32 src_tensor; in Winograd4x4To36TileX6Test() local
32 src_tensor.shape = BHWC(1, 4, 4, 1); in Winograd4x4To36TileX6Test()
33 src_tensor.data.resize(16); in Winograd4x4To36TileX6Test()
35 src_tensor.data[i] = sin(i); in Winograd4x4To36TileX6Test()
51 const int index = src_tensor.shape.LinearIndex({0, y - 1, i - 1, 0}); in Winograd4x4To36TileX6Test()
52 sum += src_tensor.data[index] * b_t[x * 6 + i]; in Winograd4x4To36TileX6Test()
87 src_tensor, in Winograd4x4To36TileX6Test()
97 TensorFloat32 src_tensor; in Winograd36To4x4Tile4x1Test() local
98 src_tensor.shape = BHWC(1, 36, 1, 1); in Winograd36To4x4Tile4x1Test()
99 src_tensor.data.resize(36); in Winograd36To4x4Tile4x1Test()
[all …]
|
| D | relu_test_util.cc | 30 TensorFloat32 src_tensor; in ReLUNoClipNoAlphaTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 2); in ReLUNoClipNoAlphaTest()
32 src_tensor.data = {-0.5f, 0.8f, -0.6f, 3.2f}; in ReLUNoClipNoAlphaTest()
49 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in ReLUNoClipNoAlphaTest()
59 TensorFloat32 src_tensor; in ReLUClipTest() local
60 src_tensor.shape = BHWC(1, 2, 1, 2); in ReLUClipTest()
61 src_tensor.data = {-0.5f, 0.8f, -0.6f, 3.2f}; in ReLUClipTest()
78 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in ReLUClipTest()
88 TensorFloat32 src_tensor; in ReLUAlphaTest() local
89 src_tensor.shape = BHWC(1, 2, 1, 2); in ReLUAlphaTest()
[all …]
|
| D | reduce_test_util.cc | 113 TensorFloat32 src_tensor; in MeanHWTest() local
114 src_tensor.shape = BHWC(1, 2, 2, 1); in MeanHWTest()
115 src_tensor.data = {1.0f, 2.0f, 3.0f, 4.0f}; in MeanHWTest()
128 CreateReduce(axis, src_tensor.shape, OperationType::MEAN, op_def, in MeanHWTest()
131 src_tensor, std::make_unique<Reduce>(std::move(operation)), in MeanHWTest()
140 TensorFloat32 src_tensor; in ReduceSumChannelsTest() local
141 src_tensor.shape = BHWC(1, 2, 1, 5); in ReduceSumChannelsTest()
142 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()
155 CreateReduce(axis, src_tensor.shape, OperationType::REDUCE_SUM, in ReduceSumChannelsTest()
158 src_tensor, std::make_unique<Reduce>(std::move(operation)), in ReduceSumChannelsTest()
[all …]
|
| D | quantize_and_dequantize_test_util.cc | 29 TensorFloat32 src_tensor; in QuantAndDequant_Dim2Bits8Test() local
30 src_tensor.shape = BHWC(1, 3, 2, 1); in QuantAndDequant_Dim2Bits8Test()
31 src_tensor.data = {0.0f, 1.0f, 0.25f, 0.50f, 0.4444444f, 0.00001f}; in QuantAndDequant_Dim2Bits8Test()
54 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in QuantAndDequant_Dim2Bits8Test()
66 TensorFloat32 src_tensor; in QuantAndDequant_Dim3Bits8_NegativeRangeTest() local
67 src_tensor.shape = BHWC(1, 3, 1, 2); in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
68 src_tensor.data = {0.0f, -0.9f, 0.25f, 0.50f, 0.4444444f, -0.00001f}; in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
91 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
102 TensorFloat32 src_tensor; in QuantAndDequant_Dim3Bits16Test() local
103 src_tensor.shape = BHWC(1, 3, 1, 2); in QuantAndDequant_Dim3Bits16Test()
[all …]
|
| D | space_to_depth_test_util.cc | 31 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x1BlockSize2Test() local
32 src_tensor.shape = BHWC(1, 2, 2, 1); in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
33 src_tensor.data = {half(1.0f), half(2.0f), half(3.0f), half(4.0f)}; in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
45 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
57 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x2BlockSize2Test() local
58 src_tensor.shape = BHWC(1, 2, 2, 2); in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
59 src_tensor.data = {half(1.4f), half(2.3f), half(3.2f), half(4.1f), in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
72 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
85 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x3BlockSize2Test() local
86 src_tensor.shape = BHWC(1, 2, 2, 3); in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
[all …]
|
| D | conv_generic_test_util.cc | 30 TensorFloat32 src_tensor; in ConvGeneric1x1SimpleWeightsTest() local
31 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvGeneric1x1SimpleWeightsTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvGeneric1x1SimpleWeightsTest()
56 src_tensor, std::make_unique<ConvGeneric>(std::move(operation)), in ConvGeneric1x1SimpleWeightsTest()
67 TensorFloat32 src_tensor; in ConvGeneric1x1Test() local
68 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvGeneric1x1Test()
69 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvGeneric1x1Test()
93 src_tensor, std::make_unique<ConvGeneric>(std::move(operation)), in ConvGeneric1x1Test()
104 TensorFloat32 src_tensor; in ConvGenericSimpleWeightsTest() local
105 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvGenericSimpleWeightsTest()
[all …]
|
| D | fully_connected_test_util.cc | 30 TensorFloat32 src_tensor; in FullyConnectedTest() local
31 src_tensor.shape = BHWC(1, 1, 1, 4); in FullyConnectedTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in FullyConnectedTest()
53 src_tensor, std::make_unique<FullyConnected>(std::move(operation)), in FullyConnectedTest()
63 TensorFloat32 src_tensor; in FullyConnectedLargeTest() local
64 src_tensor.shape = BHWC(1, 1, 1, 8); in FullyConnectedLargeTest()
65 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in FullyConnectedLargeTest()
99 src_tensor, std::make_unique<FullyConnected>(std::move(operation)), in FullyConnectedLargeTest()
114 TensorFloat32 src_tensor; in FullyConnectedExtraLargeTest() local
115 src_tensor.shape = BHWC(1, 1, 1, kInputSize); in FullyConnectedExtraLargeTest()
[all …]
|
| D | resampler_test_util.cc | 31 TensorFloat32 src_tensor; in ResamplerIdentityTest() local
32 src_tensor.shape = shape; in ResamplerIdentityTest()
33 src_tensor.data.resize(src_tensor.shape.DimensionsProduct()); in ResamplerIdentityTest()
34 for (int i = 0; i < src_tensor.data.size(); ++i) { in ResamplerIdentityTest()
35 src_tensor.data[i] = std::sin(i); in ResamplerIdentityTest()
59 {src_tensor, warp_tensor}, in ResamplerIdentityTest()
61 src_tensor.shape, &dst_tensor)); in ResamplerIdentityTest()
62 RETURN_IF_ERROR(PointWiseNear(src_tensor.data, dst_tensor.data, eps)); in ResamplerIdentityTest()
|
| D | pooling_test_util.cc | 30 TensorFloat32 src_tensor; in AveragePoolingTest() local
31 src_tensor.shape = BHWC(1, 2, 2, 2); in AveragePoolingTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in AveragePoolingTest()
52 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in AveragePoolingTest()
61 TensorFloat32 src_tensor; in AveragePoolingNonEmptyPaddingTest() local
62 src_tensor.shape = BHWC(1, 2, 2, 1); in AveragePoolingNonEmptyPaddingTest()
63 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in AveragePoolingNonEmptyPaddingTest()
83 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in AveragePoolingNonEmptyPaddingTest()
93 TensorFloat32 src_tensor; in MaxPoolingTest() local
94 src_tensor.shape = BHWC(1, 2, 2, 2); in MaxPoolingTest()
[all …]
|
| D | depthwise_conv_test_util.cc | 30 TensorFloat32 src_tensor; in DepthwiseConvSimpleWeightsTest() local
31 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConvSimpleWeightsTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in DepthwiseConvSimpleWeightsTest()
56 src_tensor, std::make_unique<DepthwiseConv>(std::move(operation)), in DepthwiseConvSimpleWeightsTest()
67 TensorFloat32 src_tensor; in DepthwiseConvNoMultiplierTest() local
68 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConvNoMultiplierTest()
69 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in DepthwiseConvNoMultiplierTest()
93 src_tensor, std::make_unique<DepthwiseConv>(std::move(operation)), in DepthwiseConvNoMultiplierTest()
104 TensorFloat32 src_tensor; in DepthwiseConvMultiplier2Test() local
105 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConvMultiplier2Test()
[all …]
|
| D | conv_buffer_1x1_test_util.cc | 29 TensorFloat32 src_tensor; in ConvBuffer1x1SimpleWeightsTest() local
30 src_tensor.shape = BHWC(1, 1, 2, 4); in ConvBuffer1x1SimpleWeightsTest()
31 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvBuffer1x1SimpleWeightsTest()
54 CreateConvBuffer1x1(env->GetGpuInfo(), op_def, attr, &src_tensor.shape); in ConvBuffer1x1SimpleWeightsTest()
56 src_tensor, absl::make_unique<ConvBuffer1x1>(std::move(operation)), in ConvBuffer1x1SimpleWeightsTest()
65 TensorFloat32 src_tensor; in ConvBuffer1x1Test() local
66 src_tensor.shape = BHWC(1, 1, 2, 4); in ConvBuffer1x1Test()
67 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvBuffer1x1Test()
91 CreateConvBuffer1x1(env->GetGpuInfo(), op_def, attr, &src_tensor.shape); in ConvBuffer1x1Test()
93 src_tensor, absl::make_unique<ConvBuffer1x1>(std::move(operation)), in ConvBuffer1x1Test()
|
| D | split_test_util.cc | 30 TensorFloat32 src_tensor; in SplitChannelsTest() local
31 src_tensor.shape = BHWC(1, 3, 2, 5); in SplitChannelsTest()
32 src_tensor.data = { in SplitChannelsTest()
54 {src_tensor}, std::make_unique<Split>(std::move(operation)), in SplitChannelsTest()
73 TensorFloat32 src_tensor; in SplitChannelsX4Test() local
74 src_tensor.shape = BHWC(1, 2, 2, 8); in SplitChannelsX4Test()
75 src_tensor.data = {half(0.1f), half(0.2f), half(0.3f), half(0.4), in SplitChannelsX4Test()
98 {src_tensor}, std::make_unique<Split>(std::move(operation)), in SplitChannelsX4Test()
118 TensorFloat32 src_tensor; in SplitWidthTest() local
119 src_tensor.shape = BHWC(1, 6, 5, 1); in SplitWidthTest()
[all …]
|
| D | elementwise_test_util.cc | 30 TensorFloat32 src_tensor; in AbsTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 2); in AbsTest()
32 src_tensor.data = {half(0.0f), half(-1.0f), half(-0.05f), half(0.045f)}; in AbsTest()
45 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in AbsTest()
56 TensorFloat32 src_tensor; in CosTest() local
57 src_tensor.shape = BHWC(1, 2, 1, 2); in CosTest()
58 src_tensor.data = {0.0f, -1.0f, -0.05f, 0.045f}; in CosTest()
72 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in CosTest()
83 TensorFloat32 src_tensor; in CopyTest() local
84 src_tensor.shape = BHWC(1, 2, 1, 2); in CopyTest()
[all …]
|
| D | reshape_test_util.cc | 31 TensorFloat32 src_tensor; in ReshapeTest() local
32 src_tensor.shape = BHWC(1, 2, 1, 3); in ReshapeTest()
33 src_tensor.data = {half(0.5f), half(-1.1f), half(-2.2f), in ReshapeTest()
46 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in ReshapeTest()
57 TensorFloat32 src_tensor; in Reshapex4Test() local
58 src_tensor.shape = BHWC(1, 1, 1, 8); in Reshapex4Test()
59 src_tensor.data = {half(0.5f), half(-1.1f), half(-2.2f), half(3.1f), in Reshapex4Test()
72 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in Reshapex4Test()
|
| D | prelu_test_util.cc | 30 TensorFloat32 src_tensor; in PReLUAlphaTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 2); in PReLUAlphaTest()
32 src_tensor.data = {0.0f, -1.0f, -2.0f, 3.0f}; in PReLUAlphaTest()
51 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in PReLUAlphaTest()
61 TensorFloat32 src_tensor; in PReLUHWCAlphaTest() local
62 src_tensor.shape = BHWC(1, 2, 1, 2); in PReLUHWCAlphaTest()
63 src_tensor.data = {0.0f, -1.0f, -2.0f, 3.0f}; in PReLUHWCAlphaTest()
82 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in PReLUHWCAlphaTest()
|
| D | one_hot_test_util.cc | 26 tflite::gpu::Tensor<BHWC, DataType::INT32> src_tensor; in OneHotTest() local
27 src_tensor.shape = BHWC(1, 1, 1, 1); in OneHotTest()
28 src_tensor.data = {3}; in OneHotTest()
40 src.UploadData(src_tensor); in OneHotTest()
55 tflite::gpu::Tensor<BHWC, DataType::INT32> src_tensor; in OneHotBatchTest() local
56 src_tensor.shape = BHWC(10, 1, 1, 1); in OneHotBatchTest()
57 src_tensor.data = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}; in OneHotBatchTest()
69 src.UploadData(src_tensor); in OneHotBatchTest()
|
| D | depthwise_conv_3x3_test_util.cc | 30 TensorFloat32 src_tensor; in DepthwiseConv3x3SimpleWeightsTest() local
31 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConv3x3SimpleWeightsTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in DepthwiseConv3x3SimpleWeightsTest()
57 src_tensor, std::make_unique<DepthwiseConv3x3>(std::move(operation)), in DepthwiseConv3x3SimpleWeightsTest()
68 TensorFloat32 src_tensor; in DepthwiseConv3x3Test() local
69 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConv3x3Test()
70 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in DepthwiseConv3x3Test()
95 src_tensor, std::make_unique<DepthwiseConv3x3>(std::move(operation)), in DepthwiseConv3x3Test()
|
| D | convolution_transposed_test_util.cc | 31 TensorFloat32 src_tensor; in ConvolutionTransposedSimpleWeightsTest() local
32 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvolutionTransposedSimpleWeightsTest()
33 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvolutionTransposedSimpleWeightsTest()
57 src_tensor, in ConvolutionTransposedSimpleWeightsTest()
72 TensorFloat32 src_tensor; in ConvolutionTransposedTest() local
73 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvolutionTransposedTest()
74 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvolutionTransposedTest()
97 src_tensor, in ConvolutionTransposedTest()
|
| D | conv_constants_test_util.cc | 30 TensorFloat32 src_tensor; in ConvConstantsSimpleWeightsTest() local
31 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvConstantsSimpleWeightsTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvConstantsSimpleWeightsTest()
56 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in ConvConstantsSimpleWeightsTest()
66 TensorFloat32 src_tensor; in ConvConstantsTest() local
67 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvConstantsTest()
68 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvConstantsTest()
93 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in ConvConstantsTest()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_aggregate_ops.cc | 42 const Tensor& src_tensor = MklGetInput(ctx, src_index); in GetTensorShape() local
46 : src_tensor.shape(); in GetTensorShape()
95 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in ComputeScalar() local
96 T* src_i = const_cast<T*>(src_tensor.flat<T>().data()); in ComputeScalar()
137 const Tensor& src_tensor = MklGetInput(ctx, kSrc0Idx); in Compute() local
142 if (src_tensor.shape().num_elements() == 0) { in Compute()
144 output_tf_shape = src_tensor.shape(); in Compute()
150 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 …]
|