Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
Dfused_conv2d_bias_activation_benchmark.py32 def build_conv_bias_relu_graph(device, input_shape, filter_shape, strides, argument
77 def build_fused_conv_bias_relu_graph(device, input_shape, filter_shape, strides, argument
134 def _run_graph(self, device, input_shape, filter_shape, strides, padding, argument
/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
247 Shape filter_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in XLA_TEST_F() local
277 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 2}); in RunTest() local
313 Shape filter_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in XLA_TEST_F() local
344 Shape filter_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in XLA_TEST_F() local
378 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 2}); in RunTest() local
416 Shape filter_shape = ShapeUtil::MakeShape(F32, filter_dims); in XLA_TEST_F() local
[all …]
/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/core/kernels/
Dconv_grad_ops.cc59 const TensorShape& filter_shape, const TensorShape& output_shape, in ConvBackpropExtractAndVerifyDimension()
100 const TensorShape& filter_shape, const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensionsV2()
161 const TensorShape& filter_shape, in ConvBackpropComputeDimensions()
Dwinograd_transform.h45 virtual const Shape& filter_shape() const { return filter_shape_; } in filter_shape() function
Dconv_grad_ops_3d.cc218 const TensorShape& filter_shape = filter.shape(); in Compute() local
324 const TensorShape& filter_shape = filter.shape(); in Compute() local
679 TensorShape filter_shape; in Compute() local
791 TensorShape filter_shape; in Compute() local
1140 const TensorShape& filter_shape = filter.shape(); in Compute() local
1529 TensorShape filter_shape; in Compute() local
Dconv_grad_filter_ops.cc245 TensorShape filter_shape; in Compute() local
498 TensorShape filter_shape; in Compute() local
560 TensorShape filter_shape = filter_backprop->shape(); in operator ()() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc96 xla::XlaOp CreateExpandedFilterMask(const xla::Shape& filter_shape, in CreateExpandedFilterMask()
128 xla::XlaOp ReshapeFilterForDepthwiseConvolution(const xla::Shape& filter_shape, in ReshapeFilterForDepthwiseConvolution()
146 xla::XlaOp ContractFilterForDepthwiseBackprop(const xla::Shape& filter_shape, in ContractFilterForDepthwiseBackprop()
204 const xla::Shape& filter_shape, const xla::Shape& out_backprop_shape, in ConvBackpropComputeDimensionsV2XlaShapes()
396 const xla::Shape& filter_shape, xla::XlaOp gradients, in MakeXlaBackpropFilterConvOp()
Dconv_ops.cc161 xla::Shape filter_shape = in Compile() local
/external/tensorflow/tensorflow/contrib/fused_conv/ops/
Dfused_conv2d_bias_activation_op.cc66 ShapeHandle filter_shape; in __anon4e6e467d0102() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc112 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
175 auto filter_shape = Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
240 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
317 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
396 auto filter_shape = ops::Placeholder::Shape({1, 1, 3, 128}); in TEST_F() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dcudnn_determinism_test.py55 def _random_out_op(self, in_shape, filter_shape): argument
Datrous_convolution_test.py98 def _test_atrous_convolution(self, add_check, input_shape, filter_shape, argument
Ddepthwise_conv_op_test.py352 filter_shape, argument
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h25 const int8_t* input_data, const RuntimeShape& filter_shape, in FullyConnected()
Dconv.h27 const int8* input_data, const RuntimeShape& filter_shape, in ConvPerChannel()
Ddepthwise_conv.h25 const int8* input_data, const RuntimeShape& filter_shape, in DepthwiseConvPerChannel()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc64 const Shape* filter_shape; member
119 const Shape& filter_shape = *params.filter_shape; in RunCudnnConvImpl() local
Dgpu_layout_assignment.cc106 Shape* filter_shape; in AddBackendConstraintsToDnnConvCustomCall() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_float.h27 const float* input_data, const RuntimeShape& filter_shape, in DepthwiseConv()
Dfully_connected.h67 const uint8* input_data, const RuntimeShape& filter_shape, in FullyConnected()
116 const uint8* input_data, const RuntimeShape& filter_shape, in FullyConnected()
Ddepthwiseconv_uint8.h180 const uint8* input_data, const RuntimeShape& filter_shape, in DepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc108 const RuntimeShape& filter_shape, const uint8* filter_data, in DispatchDepthwiseConv()
291 const std::uint8_t* filter_data, const RuntimeShape& filter_shape, in TestOneDepthwiseConvWithGivenOutputShift()
424 const std::uint8_t* filter_data, const RuntimeShape& filter_shape, in TestOneDepthwiseConvBisectOutputShift()
474 const std::uint8_t* filter_data, const RuntimeShape& filter_shape, in TestOneDepthwiseConv()
Ddepthwiseconv_float_test.cc34 const float* input_data, const RuntimeShape& filter_shape, in TestOneDepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h137 const float* input_data, const RuntimeShape& filter_shape, in Conv()

12