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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_thunk.cc 39 auto physical_dim_size = [&](int64 physical_dim) {
40 return shape.dimensions(LayoutUtil::Major(shape.layout(), physical_dim));
50 int64 physical_dim; local
51 for (physical_dim = 0; physical_dim != logical_to_physical[feature_index];
52 ++physical_dim) {
53 CHECK_LT(physical_dim, shape.dimensions_size());
54 batch_size *= physical_dim_size(physical_dim);
56 ++physical_dim; // Skip the feature dimension.
57 for (; physical_dim < shape.dimensions_size(); ++physical_dim)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.cc 274 int64 physical_dim = logical_to_physical[dimension_mapping[i]]; local
276 std::min(min_broadcasted_dimension, physical_dim);
278 std::max(max_broadcasted_dimension, physical_dim);

Completed in 390 milliseconds