Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc117 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 1, 2}); in RunTest() local
148 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 2, 2}); in RunTest() local
186 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 2, 2}); in RunTest() local
226 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 3, 3}); in RunTest() local
263 Shape filter_shape = ShapeUtil::MakeShape(F32, filter_dims); in XLA_TEST_F() local
328 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
385 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
449 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
517 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
579 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
[all …]
Dconvolution_test_1d.cc85 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in TestImpl() local
207 Shape filter_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in XLA_TEST_F() local
237 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 2}); in RunTest() local
273 Shape filter_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in XLA_TEST_F() local
304 Shape filter_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in XLA_TEST_F() local
338 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.cc125 TensorShape filter_shape = filter_backprop->shape(); in operator ()() local
332 TensorShape filter_shape; in Compute() local
441 TensorShape filter_shape; in Compute() local
686 TensorShape filter_shape = filter_backprop->shape(); in operator ()() local
Dconv_grad_ops_3d.cc231 const TensorShape& filter_shape = filter.shape(); in Compute() local
360 const TensorShape& filter_shape = filter.shape(); in Compute() local
743 TensorShape filter_shape; in Compute() local
881 TensorShape filter_shape; in Compute() local
1265 const TensorShape& filter_shape = filter.shape(); in Compute() local
1650 TensorShape filter_shape; in Compute() local
/external/tensorflow/tensorflow/python/ops/
Dconv2d_benchmark.py40 def build_graph(device, dtype, data_format, input_shape, filter_shape, strides, argument
93 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/reference/integer_ops/
Dfully_connected.h35 const int8_t* input_data, const RuntimeShape& filter_shape, in FullyConnectedPerChannel()
77 const int16_t* input_data, const RuntimeShape& filter_shape, in FullyConnectedPerChannel()
115 const int8_t* input_data, const RuntimeShape& filter_shape, in FullyConnected()
159 const int16_t* input_data, const RuntimeShape& filter_shape, in FullyConnected()
Ddepthwise_conv.h27 const int8_t* input_data, const RuntimeShape& filter_shape, in DepthwiseConvPerChannel()
127 const int16_t* input_data, const RuntimeShape& filter_shape, in DepthwiseConvPerChannel()
213 const RuntimeShape& filter_shape, const int8_t* filter_data, in DepthwiseConvHybridPerChannel()
Dconv.h29 const int8_t* input_data, const RuntimeShape& filter_shape, in ConvPerChannel()
142 const int16_t* input_data, const RuntimeShape& filter_shape, in ConvPerChannel()
Dtranspose_conv.h29 const int8_t* input_data, const RuntimeShape& filter_shape, in TransposeConv()
129 const int16_t* input_data, const RuntimeShape& filter_shape, in TransposeConv()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dreplace_cast_hacks_with_tf_xla_ops.cc111 ShapedType filter_shape = filter.getType().template cast<ShapedType>(); in CreateXLAConvOp() local
168 ShapedType filter_shape = filter.getType().template cast<ShapedType>(); in CreateXLAConvOpFromTFConv2DOp() local
188 ShapedType filter_shape = filter.getType().template cast<ShapedType>(); in CreateXLAConvOpFromTFDepthwiseConv2DOp() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h45 const int8* input_data, const RuntimeShape& filter_shape, in DepthwiseConvHybridGeneral()
264 const RuntimeShape& filter_shape, const int8* filter_data, in DepthwiseConvHybridWithRounding()
323 const RuntimeShape& filter_shape, const int8* filter_data, in DepthwiseConvHybridImpl()
381 const RuntimeShape& filter_shape; member
397 const RuntimeShape& filter_shape, const int8* filter_data, in DepthwiseConvHybridPerChannel()
Dfully_connected.h34 const InputScalar* input_data, const RuntimeShape& filter_shape, in FullyConnectedPerChannel()
103 const InputScalar* input_data, const RuntimeShape& filter_shape, in FullyConnected()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h82 const RuntimeShape& filter_shape) { in HowManyConvThreads()
123 const RuntimeShape& filter_shape, in DepthwiseConv()
/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/core/grappler/optimizers/
Dremapper_test.cc638 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in RunTest() local
745 auto filter_shape = Placeholder::Shape({1, 1, 3, 128}); in RunTest() local
899 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in RunTest() local
1022 auto filter_shape = ops::Placeholder::Shape({1, 1, 1, 3, 6}); in TEST_F() local
1087 auto filter_shape = ops::Placeholder::Shape({1, 1, 1, 3, 6}); in TEST_F() local
1153 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 6}); in TEST_F() local
1283 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in RunTest() local
1521 auto filter_shape = Placeholder::Shape({3, 3, 3, 128}); in TEST_F() local
1728 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
1807 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
[all …]
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dconv_bias_activation.py75 def add_bias_add(data_input, filter_shape): argument
/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/compiler/mlir/tosa/transforms/
Dfuse_bias_tf.cc85 auto filter_shape = in matchAndRewrite() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv.h28 const float* input_data, const RuntimeShape& filter_shape, in Conv()
111 const uint8_t* input_data, const RuntimeShape& filter_shape, in Conv()
206 const RuntimeShape& filter_shape, const int8_t* filter_data, in HybridConvPerChannel()
Dtranspose_conv.h29 const float* input_data, const RuntimeShape& filter_shape, in TransposeConv()
118 const uint8_t* input_data, const RuntimeShape& filter_shape, in TransposeConv()
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dconv_spmd_expander.cc99 llvm::ArrayRef<int64_t> filter_shape = filter_type.getShape(); in VerifyConvLayout() local
173 auto filter_shape = filter_type.getShape(); in HandleConv() local
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
Dtf_to_xla_attribute_utils.cc93 ShapedType filter_shape = filter.getType().template cast<ShapedType>(); in PadForDynamicShapedInputSamePadding() local
150 ShapedType filter_shape = filter.getType().template cast<ShapedType>(); in CalculatePaddingAndPadIfNeeded() local

1234