Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc55 const TensorShape& filter_shape, const TensorShape& output_shape, in ConvBackpropExtractAndVerifyDimension()
96 const TensorShape& filter_shape, const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensionsV2()
157 const TensorShape& filter_shape, in ConvBackpropComputeDimensions()
170 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.cc121 TensorShape filter_shape = filter_backprop->shape(); in operator ()() local
328 TensorShape filter_shape; in Compute() local
437 TensorShape filter_shape; in Compute() local
667 TensorShape filter_shape = filter_backprop->shape(); in operator ()() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc118 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 1, 2}); in RunTest() local
149 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 2, 2}); in RunTest() local
183 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 2, 2}); in RunTest() local
219 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 3, 3}); in RunTest() local
248 Shape filter_shape = ShapeUtil::MakeShape(F32, filter_dims); in XLA_TEST_F() local
314 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
373 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
439 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
509 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest() local
573 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/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 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/micro/kernels/arc_mli/
Ddepthwise_conv_slicing_test.cc200 const int filter_shape[] = {4, 2, 3, 3, 2}; in TF_LITE_MICRO_TEST() local
237 const int filter_shape[] = {4, 2, 3, 3, 2}; in TF_LITE_MICRO_TEST() local
286 const int filter_shape[] = {4, 2, 3, 3, 2}; in TF_LITE_MICRO_TEST() local
327 const int filter_shape[] = {4, 2, 3, 3, 2}; in TF_LITE_MICRO_TEST() local
381 const int filter_shape[] = {4, 1, 3, 3, 10}; in TF_LITE_MICRO_TEST() local
419 const int filter_shape[] = {4, 1, 3, 3, 10}; in TF_LITE_MICRO_TEST() local
470 const int filter_shape[] = {4, 1, 3, 3, 10}; in TF_LITE_MICRO_TEST() local
508 const int filter_shape[] = {4, 1, 3, 3, 10}; in TF_LITE_MICRO_TEST() local
/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/micro/kernels/
Ddepthwise_conv_test.cc246 const int filter_shape[] = {4, 1, 2, 2, 4}; in TF_LITE_MICRO_TEST() local
273 const int filter_shape[] = {4, 1, 2, 2, 4}; in TF_LITE_MICRO_TEST() local
319 const int filter_shape[] = {4, 1, 2, 2, 4}; in TF_LITE_MICRO_TEST() local
361 const int filter_shape[] = {4, 1, 2, 2, 4}; in TF_LITE_MICRO_TEST() local
386 const int filter_shape[] = {4, 1, 2, 2, 4}; in TF_LITE_MICRO_TEST() local
432 const int filter_shape[] = {4, 2, 1, 8, 1}; in TF_LITE_MICRO_TEST() local
470 const int filter_shape[] = {4, 1, 2, 2, 4}; in TF_LITE_MICRO_TEST() local
511 const int filter_shape[] = {4, 1, 2, 2, 2}; in TF_LITE_MICRO_TEST() local
555 const int filter_shape[] = {4, 1, 2, 2, 4}; in TF_LITE_MICRO_TEST() local
598 const int filter_shape[] = {4, 1, 2, 2, 4}; in TF_LITE_MICRO_TEST() local
[all …]
/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()
/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/micro/kernels/cmsis_nn/
Dfully_connected.cc75 RuntimeShape filter_shape = GetTensorShape(filter); in Prepare() local
112 const RuntimeShape filter_shape = tflite::micro::GetTensorShape(filter); in EvalQuantizedInt8() local
Ddepthwise_conv.cc167 RuntimeShape filter_shape = GetTensorShape(filter); in Prepare() local
270 RuntimeShape filter_shape = tflite::micro::GetTensorShape(filter); in EvalQuantizedPerChannel() local
386 RuntimeShape filter_shape = tflite::micro::GetTensorShape(filter); in EvalQuantized() local
/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/python/kernel_tests/
Dcudnn_deterministic_base.py74 def _random_out_op(self, in_shape, filter_shape, strides, padding): argument
/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()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dfuse_bias_tf.cc84 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
480 auto filter_shape = Placeholder::Shape({3, 3, 3, 128}); in TEST_F() local
551 auto filter_shape = Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
759 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
838 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
943 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dconv.cc38 const RuntimeShape& filter_shape, const int8_t* filter_data, in ConvPerChannel()
173 const RuntimeShape& filter_shape, const int8_t* filter_data, in Conv1x32Input32x32Filter()

1234