Home
last modified time | relevance | path

Searched refs:conv3d (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/python/layers/
Dconvolutional.py29 conv3d = convolutional.conv3d variable
49 convolution3d = conv3d
Dlayers.py58 from tensorflow.python.layers.convolutional import conv3d
/external/tensorflow/tensorflow/lite/kernels/
Dconv3d.cc32 namespace conv3d { namespace
157 static TfLiteRegistration r = {conv3d::Init, conv3d::Free, conv3d::Prepare, in Register_CONV_3D()
158 conv3d::Eval}; in Register_CONV_3D()
DAndroid.bp78 "conv3d.cc",
/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py793 y = gen_nn_ops.conv3d(x, w, strides, 'SAME')
1247 y = gen_nn_ops.conv3d(x, w, [1, 1, 1, 1, 1], 'SAME')
1277 conv3d = gen_nn_ops.conv3d(x, w, [1, 1, 1, 1, 1], 'SAME')
1278 y = math_ops.reduce_mean(conv3d, [0, 1, 2, 3], keepdims=True)
1314 conv3d = gen_nn_ops.conv3d(x, w, [1, 1, 1, 1, 1], 'SAME')
1316 conv3d,
1357 conv3d = gen_nn_ops.conv3d(x_3d, filters, strides_val, 'SAME')
1358 y, _, _ = nn.fused_batch_norm(conv3d, scale, offset, data_format='NDHWC')
1392 conv3d = gen_nn_ops.conv3d(x_3d, filters, strides_val, 'SAME')
1394 conv3d,
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_3d_test.py93 conv = nn_ops.conv3d(t1, t2, strides, padding=padding,
157 computed = nn_ops.conv3d(
401 conv = nn_ops.conv3d(
525 conv = nn_ops.conv3d(
809 actual = nn_ops.conv3d(
Dconv3d_backprop_filter_v2_grad_test.py48 conv_out = nn_ops.conv3d(in_val,
Dcudnn_deterministic_base.py106 out_op = nn_ops.conv3d(
Dinit_ops_test.py898 convolution = convolutional.conv3d
1291 convolution = convolutional.conv3d
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional.py541 def conv3d(inputs, function
1509 convolution3d = conv3d
Dconvolutional_test.py337 conv3d = conv_layers.Conv3D(2, 3,
341 conv3d(inputs)
342 self.assertEqual(conv3d.kernel_constraint, k_constraint)
343 self.assertEqual(conv3d.bias_constraint, b_constraint)
350 conv_layers.conv3d(images, 32, 9, data_format='channels_first')
/external/tensorflow/tensorflow/compiler/tests/
Dconv3d_test.py50 conv_out = nn_ops.conv3d(in_val,
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp57 "kernels/conv3d.cc",
/external/tensorflow/tensorflow/python/keras/
Dbackend_test.py1152 y = backend.conv3d(x, k, padding='valid', data_format='channels_first')
1157 y = backend.conv3d(
1163 y = backend.conv3d(
1169 y = backend.conv3d(
1173 y = backend.conv3d(
1176 y = backend.conv3d(x, k, (2, 2, 2), data_format='other')
1178 y = backend.conv3d(x, k, (2, 2))
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.layers.pbtxt112 name: "conv3d"
Dtensorflow.keras.backend.pbtxt104 name: "conv3d"
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py3010 return gen_nn_ops.conv3d(
3022 gen_nn_ops.conv3d,
3059 return gen_nn_ops.conv3d(
3064 gen_nn_ops.conv3d.__doc__, "filter", "filters")
3065 conv3d_v1.__doc__ = gen_nn_ops.conv3d.__doc__
Dnn_grad.py186 nn_ops.conv3d(
208 nn_ops.conv3d(
/external/tensorflow/tensorflow/lite/kernels/internal/
DBUILD462 "reference/conv3d.h",
581 "reference/conv3d.h",
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc261 csinfo_.conv3d = "Conv3D"; in MklLayoutRewritePass()
434 {csinfo_.conv3d, mkl_op_registry::GetMklOpName(csinfo_.conv3d), in MklLayoutRewritePass()
768 std::bind(CheckForMklOp, std::placeholders::_1, csinfo_.conv3d); in MklLayoutRewritePass()
921 string conv3d; member
/external/tensorflow/tensorflow/lite/python/
Dlite_v2_test.py499 conv3d = tf.nn.conv3d(
504 erf = tf.math.erf(conv3d)
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dslice_op_test.py174 conv_t = nn_ops.conv3d(
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt84 name: "conv3d"
Dtensorflow.keras.backend.pbtxt100 name: "conv3d"
/external/llvm-project/mlir/include/mlir/Dialect/Tosa/IR/
DTosaOps.td111 // Operator: conv3d
113 def Tosa_Conv3DOp : Tosa_Op<"conv3d", [NoSideEffect]> {

12