HomeSort by relevance Sort by last modified time
    Searched refs:Conv3DOp (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
conv_ops_3d.cc 66 class Conv3DOp : public BinaryOp<T> {
68 explicit Conv3DOp(OpKernelConstruction* context) : BinaryOp<T>(context) {
147 Conv3DOp<CPUDevice, T>);
498 Conv3DOp<GPUDevice, Eigen::half>);
501 Conv3DOp<GPUDevice, float>);
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
conv_ops.cc 312 class Conv3DOp : public ConvOp {
314 explicit Conv3DOp(OpKernelConstruction* ctx)
317 REGISTER_XLA_OP(Name("Conv3D"), Conv3DOp);

Completed in 206 milliseconds