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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
conv_ops.cc 505 const int c_dim = GetTensorFeatureDimIndex(num_dims(), data_format_); variable
508 ctx, (strides_[n_dim] == 1 && strides_[c_dim] == 1),
517 ctx, dilations_[n_dim] == 1 && dilations_[c_dim] == 1,
561 // Swap n_dim and c_dim in the activations.
562 dnums.set_input_batch_dimension(c_dim);
569 dnums.set_kernel_output_feature_dimension(c_dim);

Completed in 163 milliseconds