Searched defs:in_planes (Results 1 – 9 of 9) sorted by relevance
| /external/pytorch/test/fx/ |
| D | test_z3_gradual_types.py | 1183 in_planes, argument
1219 in_planes, argument
1290 in_planes, argument
1326 in_planes, argument
1364 in_planes, argument
1400 in_planes, argument
1478 in_planes, argument
1607 in_planes, argument
2039 in_planes, argument
2356 in_planes, argument
|
| D | test_gradual_type.py | 32 def conv3x3(in_planes, out_planes, stride=1, groups=1, dilation=1): argument
513 in_planes, argument
561 in_planes, argument
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_volume_patches_op.cc | 79 const int in_planes = input.dim_size(1); in Compute() local
|
| /external/pytorch/benchmarks/functional_autograd_benchmark/ |
| D | torchvision_models.py | 19 def conv3x3(in_planes, out_planes, stride=1, groups=1, dilation=1): argument
33 def conv1x1(in_planes, out_planes, stride=1): argument
|
| /external/pytorch/test/jit/ |
| D | test_models.py | 618 def conv1x1(in_planes, out_planes, stride=1): argument
624 def conv3x3(in_planes, out_planes, stride=1): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 241 int64_t in_planes = GetTensorDim(input, data_format, '0'); in launch() local
|
| D | fused_batch_norm_op.cc | 1298 int64_t in_planes = GetTensorDim(x, tensor_format_, '0'); in ComputeWithReservedSpace() local
1543 int64_t in_planes = GetTensorDim(x, tensor_format_, '0'); in ComputeWithReservedSpace() local
|
| D | pooling_ops_3d.cc | 598 const int32_t in_planes = params.tensor_in_planes; in launch() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2779 auto in_planes = c->Value(in_planes_dim); in __anon847f0b683a02() local
|