Home
last modified time | relevance | path

Searched defs:filter_shape (Results 1 – 25 of 84) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc122 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 1, 2}); in RunTest() local
153 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 2, 2}); in RunTest() local
191 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 2, 2}); in RunTest() local
231 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 3, 3}); in RunTest() local
268 Shape filter_shape = ShapeUtil::MakeShape(F32, filter_dims); in XLA_TEST_F() local
334 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
393 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
459 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
529 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
593 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
[all …]
Dconvolution_test_1d.cc87 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in TestImpl() local
211 Shape filter_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in XLA_TEST_F() local
241 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 2}); in RunTest() local
277 Shape filter_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in XLA_TEST_F() local
308 Shape filter_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in XLA_TEST_F() local
342 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 2}); in RunTest() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc55 const TensorShape& filter_shape, const TensorShape& output_shape, in ConvBackpropExtractAndVerifyDimension()
97 const TensorShape& filter_shape, const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensionsV2()
162 const TensorShape& filter_shape, in ConvBackpropComputeDimensions()
175 const TensorShape& filter_shape, in Conv2DBackpropComputeInputShape()
Dwinograd_transform.h45 virtual const Shape& filter_shape() const { return filter_shape_; } in filter_shape() function
Dconv_grad_filter_ops.cc124 TensorShape filter_shape = filter_backprop->shape(); in operator ()() local
331 TensorShape filter_shape; in Compute() local
440 TensorShape filter_shape; in Compute() local
684 TensorShape filter_shape = filter_backprop->shape(); in operator ()() local
Dconv_grad_ops_3d.cc229 const TensorShape& filter_shape = filter.shape(); in Compute() local
358 const TensorShape& filter_shape = filter.shape(); in Compute() local
741 TensorShape filter_shape; in Compute() local
875 TensorShape filter_shape; in Compute() local
1254 const TensorShape& filter_shape = filter.shape(); in Compute() local
1803 TensorShape filter_shape; in Compute() local
/external/tensorflow/tensorflow/python/ops/
Dconv2d_benchmark.py44 def build_graph(device, dtype, data_format, input_shape, filter_shape, strides, argument
97 def _run_graph(self, device, dtype, data_format, input_shape, filter_shape, argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc49 const xla::Shape& filter_shape) { in GroupedFilterShapeForDepthwiseConvolution()
65 xla::XlaOp filter, const xla::Shape& filter_shape, int64_t num_groups, in TransposeFilterForGroupConvolutionBackpropInput()
90 xla::XlaOp ReshapeFilterForDepthwiseConvolution(const xla::Shape& filter_shape, in ReshapeFilterForDepthwiseConvolution()
138 const xla::Shape& filter_shape, const xla::Shape& out_backprop_shape, in ConvBackpropComputeDimensionsV2XlaShapes()
393 const xla::Shape& filter_shape, xla::XlaOp gradients, in MakeXlaBackpropFilterConvOp()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h43 const int8* input_data, const RuntimeShape& filter_shape, in DepthwiseConvHybridGeneral()
262 const RuntimeShape& filter_shape, const int8* filter_data, in DepthwiseConvHybridWithRounding()
321 const RuntimeShape& filter_shape, const int8* filter_data, in DepthwiseConvHybridImpl()
379 const RuntimeShape& filter_shape; member
395 const RuntimeShape& filter_shape, const int8* filter_data, in DepthwiseConvHybridPerChannel()
Dfully_connected.h32 const int8* input_data, const RuntimeShape& filter_shape, in FullyConnected()
Dconv.h34 const int8* input_data, const RuntimeShape& filter_shape, in ConvPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h25 const int8_t* input_data, const RuntimeShape& filter_shape, in FullyConnected()
67 const int16_t* input_data, const RuntimeShape& filter_shape, in FullyConnected()
Ddepthwise_conv.h25 const int8_t* input_data, const RuntimeShape& filter_shape, in DepthwiseConvPerChannel()
125 const int16_t* input_data, const RuntimeShape& filter_shape, in DepthwiseConvPerChannel()
211 const RuntimeShape& filter_shape, const int8_t* filter_data, in DepthwiseConvHybridPerChannel()
Dconv.h27 const int8_t* input_data, const RuntimeShape& filter_shape, in ConvPerChannel()
132 const int16_t* input_data, const RuntimeShape& filter_shape, in ConvPerChannel()
Dtranspose_conv.h27 const int8_t* input_data, const RuntimeShape& filter_shape, in TransposeConv()
126 const int16_t* input_data, const RuntimeShape& filter_shape, in TransposeConv()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc124 const RuntimeShape& filter_shape, in DispatchDepthwiseConvGeneral()
142 const int8* input_data, const RuntimeShape& filter_shape, in DispatchDepthwiseConvGeneral()
160 const RuntimeShape& filter_shape, in DispatchDepthwiseConvImpl()
363 const RuntimeShape& filter_shape, in DispatchDepthwiseConvImpl()
530 const RuntimeShape& filter_shape, in DispatchDepthwiseConv()
617 const RuntimeShape& filter_shape, std::int32_t filter_offset, in TestOneDepthwiseConvWithGivenOutputShift()
754 const RuntimeShape& filter_shape, std::int32_t filter_offset, in TestOneDepthwiseConvBisectOutputShift()
813 const RuntimeShape& filter_shape, std::int32_t filter_offset, in TestOneDepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h80 const RuntimeShape& filter_shape) { in HowManyConvThreads()
121 const RuntimeShape& filter_shape, in DepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv.h26 const float* input_data, const RuntimeShape& filter_shape, in Conv()
102 const uint8_t* input_data, const RuntimeShape& filter_shape, in Conv()
190 const RuntimeShape& filter_shape, const int8_t* filter_data, in HybridConvPerChannel()
Dtranspose_conv.h27 const float* input_data, const RuntimeShape& filter_shape, in TransposeConv()
116 const uint8_t* input_data, const RuntimeShape& filter_shape, in TransposeConv()
Ddepthwiseconv_float.h27 const float* input_data, const RuntimeShape& filter_shape, in DepthwiseConv()
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/
Dconv_bias_activation.py79 def add_bias_add(data_input, filter_shape): argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dcudnn_deterministic_base.py74 def _random_out_op(self, in_shape, filter_shape, strides, padding): argument
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dfuse_bias_tf.cc85 auto filter_shape = in matchAndRewrite() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc337 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
548 auto filter_shape = Placeholder::Shape({3, 3, 3, 128}); in TEST_F() local
619 auto filter_shape = Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
826 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
905 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
1009 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
Dmkl_remapper_test.cc60 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in FuseConv2DWithBiasAndAddNOrAdd() local
456 auto filter_shape = ops::Placeholder::Shape({32, 8}); in TEST_F() local

1234