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

  /external/tensorflow/tensorflow/core/util/
tensor_format.cc 42 case FORMAT_NCHW_VECT_C:
74 *format = FORMAT_NCHW_VECT_C;
tensor_format.h 47 FORMAT_NCHW_VECT_C = 2,
92 if (format == FORMAT_NCHW_VECT_C) {
111 if (format == FORMAT_NCHW_VECT_C) {
134 case FORMAT_NCHW_VECT_C:
150 case FORMAT_NCHW_VECT_C:
160 DCHECK_EQ(format, FORMAT_NCHW_VECT_C);
173 case FORMAT_NCHW_VECT_C:
263 } else if (format == FORMAT_NCHW || format == FORMAT_NCHW_VECT_C) {
408 // operations. If format is FORMAT_NCHW_VECT_C, the output TensorShape has rank
421 if (format == FORMAT_NCHW_VECT_C) {
    [all...]
tensor_format_test.cc 31 EnumStringPair(FORMAT_NCHW_VECT_C),
109 format == FORMAT_NCHW_VECT_C) ? DimMaps::kTdmNCHW[num_spatial_dims]
  /external/tensorflow/tensorflow/core/kernels/
depthtospace_op.cc 71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)),
spacetodepth_op.cc 71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)),
pooling_ops_common.cc 47 (data_format == FORMAT_NCHW_VECT_C ? 4 : 1);
depthwise_conv_op_gpu.cu.cc 616 case FORMAT_NCHW_VECT_C:
617 LOG(ERROR) << "FORMAT_NCHW_VECT_C is not supported";
693 case FORMAT_NCHW_VECT_C:
694 LOG(ERROR) << "FORMAT_NCHW_VECT_C is not supported";
    [all...]
pooling_ops_common.h 299 data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW_VECT_C,
maxpooling_op.cc     [all...]
  /external/tensorflow/tensorflow/core/framework/
common_shape_fns.cc 282 if (tensor_format == FORMAT_NCHW_VECT_C) {
303 if (format == FORMAT_NCHW_VECT_C) {
331 if (format == FORMAT_NCHW_VECT_C) {
356 if (format == tensorflow::FORMAT_NCHW_VECT_C) {
637 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4;
787 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4;
    [all...]
  /external/tensorflow/tensorflow/contrib/fused_conv/kernels/
fused_conv2d_bias_activation_op.cc 123 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)),
  /external/tensorflow/tensorflow/compiler/tests/
randomized_tests.cc 622 case FORMAT_NCHW_VECT_C:
623 LOG(FATAL) << "FORMAT_NCHW_VECT_C not supported.";
    [all...]

Completed in 463 milliseconds