Home
last modified time | relevance | path

Searched defs:dim_nums (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter_internal.h41 DotDimensionNumbers dim_nums; member
Ddot_op_emitter.cc60 DotDimensionNumbers dim_nums; member
411 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm() local
689 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_dimension_numbers_test.cc105 ConvolutionDimensionNumbers dim_nums = in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers.cc222 const DotDimensionNumbers& dim_nums = instruction->dot_dimension_numbers(); in MatchAndExplain() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc194 const DotDimensionNumbers& dim_nums = in AddBackendConstraints() local
Dgemm_thunk.cc323 DotDimensionNumbers dim_nums = GetDimensionNumbers(*hlo_instruction()); in ExecuteOnStream() local