HomeSort by relevance Sort by last modified time
    Searched full:allow_legacy_scalars (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
concat_op.cc 76 (allow_legacy_scalars() && concat_dim == 0),
119 // TODO(irving): Remove check once !allow_legacy_scalars().
124 // TODO(irving): Remove rank 0 case once !allow_legacy_scalars().
pad_op.cc 65 (allow_legacy_scalars() && dims == 0 && in1.dim_size(0) == 1) ? 1
94 (allow_legacy_scalars() && d == in0.dims()) ? 1 : in0.dim_size(d);
117 // change flat to tensor. That is, once !allow_legacy_scalars().
mkl_concat_op.cc 89 (allow_legacy_scalars() && concat_dim == 0),
132 // TODO(irving): Remove check once !allow_legacy_scalars().
137 // TODO(irving): Remove rank 0 case once !allow_legacy_scalars().
177 (allow_legacy_scalars() && concat_dim == 0),
    [all...]
quantized_concat_op.cc 189 (allow_legacy_scalars() && concat_dim == 0),
conv_ops_fused.cc 688 (allow_legacy_scalars() && dims == 0 && paddings.dim_size(0) == 1)
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
mirror_pad_op.cc 65 (allow_legacy_scalars() && dims == 0 && pad_shape.dim_size(0) == 1)
pad_op.cc 40 (allow_legacy_scalars() && dims == 0 && pad_shape.dim_size(0) == 1)
concat_op.cc 67 (allow_legacy_scalars() && concat_dim == 0),
  /external/tensorflow/tensorflow/core/framework/
op_kernel.h 147 bool allow_legacy_scalars() const { function in class:tensorflow::OpKernel
157 return shape.dims() == 0 || (allow_legacy_scalars() && shape.dims() == 1 &&
163 return shape.dims() == 1 || (allow_legacy_scalars() && shape.dims() == 0);
167 // TODO(irving): Move to TensorShapeUtils once !allow_legacy_scalars
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 77 // See OpKernel::IsLegacyScalar and OpKernel::allow_legacy_scalars.
308 // See OpKernel::IsLegacyScalar and OpKernel::allow_legacy_scalars.
    [all...]

Completed in 442 milliseconds