Home
last modified time | relevance | path

Searched defs:conv3d (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/layers/
H A Dconvolutional.py25 conv3d = convolutional.conv3d variable
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dconv.h118 inline Tensor conv3d( in conv3d() function
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dandroid_api_module.py116 def conv3d(self, x: Tensor, w: Tensor, toChannelsLast: bool) -> Tensor: member in AndroidAPIModule
/aosp_15_r20/external/pytorch/android/pytorch_android/
H A Dgenerate_test_torchscripts.py130 def conv3d(self, x: Tensor, w: Tensor, toChannelsLast: bool) -> Tensor: member in Test
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dgraph_rewrite_helper.cpp105 std::string conv3d = R"( in replaceConvolutionWithAtenConv() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dconv3d.cc35 namespace conv3d { namespace
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DConv3D.cpp193 SimpleTensor<T> conv3d(const SimpleTensor<T> &src, const SimpleTensor<T> &weights, const SimpleTens… in conv3d() function
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/
H A Dfunctional.py317 def conv3d( function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h375 std::string conv3d = R"( in quant_fusion_pattern_and_replacements() local
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
H A Dconvolutional.py543 def conv3d(inputs, function
/aosp_15_r20/external/pytorch/torch/jit/
H A D_shape_functions.py914 def conv3d( function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc355 bool IsCpuCompatibleConv3D(const RemapperContext& ctx, const NodeDef* conv3d) { in IsCpuCompatibleConv3D()
2083 void CopyConv3DAttributes(const NodeDef& conv3d, NodeDef* fused_conv3d, in CopyConv3DAttributes()
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py145 conv3d = _add_docstr( variable
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dbackend.py5603 def conv3d(x, function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc871 string conv3d; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc6478 auto conv3d = local
6485 auto conv3d = ops::Conv3D(s.WithOpName("my_conv3d"), input, filter, strides, local
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py2496 def conv3d( function