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

  /external/tensorflow/tensorflow/compiler/xla/service/
dot_decomposer.cc 34 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers();
169 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers();
hlo.proto 125 xla.DotDimensionNumbers dot_dimension_numbers = 30;
algebraic_simplifier.cc 675 dot->dot_dimension_numbers().lhs_contracting_dimensions(0);
683 dot->dot_dimension_numbers().rhs_contracting_dimensions(0);
782 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers();
959 DotDimensionNumbers dot_dimension_numbers; local
1980 DotDimensionNumbers dot_dimension_numbers; local
    [all...]
hlo_cost_analysis.cc 204 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers();
hlo_instruction.h 1106 const DotDimensionNumbers& dot_dimension_numbers() const { function in class:xla::HloInstruction
    [all...]
hlo_verifier.cc 71 dot->dot_dimension_numbers()));
    [all...]
hlo_instruction.cc 123 MakeUnique<DotDimensionNumbers>(proto.dot_dimension_numbers());
    [all...]
hlo_evaluator.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc     [all...]
ir_emitter.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_unnested.cc 294 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers();
    [all...]

Completed in 664 milliseconds