| /external/tensorflow/tensorflow/lite/kernels/ |
| D | depthwise_conv_test.cc | 481 const int filter_depth = 4; in BatchPaddingValidTest() local
487 {filter_batch, filter_size, filter_size, filter_depth}}, in BatchPaddingValidTest()
542 const int filter_depth = 1; in BatchPaddingSameTest() local
548 {filter_batch, filter_size, filter_size, filter_depth}}, in BatchPaddingSameTest()
1075 const int filter_depth = 8; in TEST_P() local
1084 {filter_batch, filter_size, filter_size, filter_depth}, in TEST_P()
1148 const int filter_depth = 8; in TEST_P() local
1157 {filter_batch, filter_size, filter_size, filter_depth}, in TEST_P()
1223 const int filter_depth = 8; in TEST_P() local
1232 {filter_batch, filter_size, filter_size, filter_depth}, in TEST_P()
[all …]
|
| D | padding.h | 84 int filter_width, int filter_depth, TfLitePadding padding, int* out_height, in ComputePadding3DValues() argument
90 *out_depth = ComputeOutSize(padding, in_depth, filter_depth, stride_depth, in ComputePadding3DValues()
97 filter_depth, *out_depth, &offset); in ComputePadding3DValues()
|
| D | conv3d.cc | 89 int filter_depth = filter->dims->data[0]; in Prepare() local
99 filter_width, filter_depth, params->padding, &out_height, &out_width, in Prepare()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | conv3d.h | 46 const int filter_depth = filter_shape.Dims(0); in Conv3D() local
64 for (int filter_d = 0; filter_d < filter_depth; ++filter_d) { in Conv3D()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 142 const int64 filter_depth = filter.dim_size(3); in Compute() local
145 OP_REQUIRES(context, in_depth % filter_depth == 0, in Compute()
148 in_depth, " vs ", filter_depth)); in Compute()
231 const int64 filter_depth = filter.dim_size(3); in launch() local
248 bool is_grouped_convolution = filter_depth != in_depth; in launch()
415 .set_input_feature_map_count(filter_depth) in launch()
428 .set_group_count(in_depth / filter_depth); in launch()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | conv.cc | 60 const int filter_depth = filter_shape.Dims(3); in ConvPerChannel() local
112 filter_depth + in ConvPerChannel()
|
| D | depthwise_conv.cc | 94 const int filter_depth = filter_shape.Dims(3); in DepthwiseConvPerChannel() local
137 ((filter_y)*filter_width + filter_x) * filter_depth + in DepthwiseConvPerChannel()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Conv3DBackpropFilterV2.pbtxt | 14 `[filter_depth, filter_height, filter_width, in_channels, out_channels]`
|
| D | api_def_Conv3D.pbtxt | 12 Shape `[filter_depth, filter_height, filter_width, in_channels,
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | ceva_tflm_lib.h | 143 const int filter_height, const int filter_width, const int filter_depth,
174 const int filter_width, const int filter_depth, const int8_t* filter_data,
187 const int filter_width, const int filter_depth, const float* filter_data,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 2332 Arg<TF_FloatTensor, [{Shape `[filter_depth, filter_height, filter_width, in_channels,
2369 `[filter_depth, filter_height, filter_width, in_channels, out_channels]`
|