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

  /external/tensorflow/tensorflow/compiler/xla/service/
transpose_folding.cc 117 const auto& transpose_dimensions = transpose.dimensions(); local
124 transpose_dimensions[dnums.input_batch_dimension()]);
126 transpose_dimensions[dnums.input_feature_dimension()]);
129 input_spatial_dimension = transpose_dimensions[input_spatial_dimension];
141 const auto& transpose_dimensions = transpose.dimensions(); local
148 transpose_dimensions[dnums.kernel_input_feature_dimension()]);
150 transpose_dimensions[dnums.kernel_output_feature_dimension()]);
153 kernel_spatial_dimension = transpose_dimensions[kernel_spatial_dimension];
algebraic_simplifier.cc 1627 auto transpose_dimensions = arg->dimensions(); local
    [all...]

Completed in 76 milliseconds