Searched refs:conv_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding_test.cc | 243 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
247 EXPECT_IS_OK(conv_shape); in TEST_F()
249 conv_shape.ValueOrDie(), x, transpose_y, in TEST_F()
300 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
304 EXPECT_IS_OK(conv_shape); in TEST_F()
306 conv_shape.ValueOrDie(), x, transpose_y, in TEST_F()
362 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
366 EXPECT_IS_OK(conv_shape); in TEST_F()
368 conv_shape.ValueOrDie(), transpose_x, y, in TEST_F()
430 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | depthwise_conv_plus_1x1_conv.cc | 238 const auto conv_shape = conv_attr.weights.shape; in IsDepthwiseConvPlus1x1ConvSupported() local
241 conv_shape.w == 1 && conv_shape.h == 1 && conv_attr.dilations.w == 1 && in IsDepthwiseConvPlus1x1ConvSupported()
249 conv_shape.o <= 32 && conv_shape.i * conv_shape.o <= 16 * 32; in IsDepthwiseConvPlus1x1ConvSupported()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | conv_test.mm | 246 BHWC conv_shape;
247 conv_shape.b = dst_shape.b;
248 conv_shape.h = 36;
249 conv_shape.w = DivideRoundUp(new_width, 4) * DivideRoundUp(new_height, 4);
250 conv_shape.c = dst_shape.c;
280 CreateConvolutionMetalWino4x4To6x6(op_def, conv_shape, attr, env->GetGpuInfo());
288 BHWC output1_shape = conv_shape;
294 BHWC output2_shape = conv_shape;
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 3500 const auto& conv_shape = conv_props[0].shape(); in MulConvPushDown() local
3501 if (!ShapesSymbolicallyEqual(mul_shape, conv_shape)) { in MulConvPushDown()
|