HomeSort by relevance Sort by last modified time
    Searched defs:broadcast_dimensions (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_instruction.cc 666 tensorflow::gtl::ArraySlice<int64> broadcast_dimensions) {
670 instruction->dimensions_.assign(broadcast_dimensions.begin(),
671 broadcast_dimensions.end());
695 std::vector<int64> broadcast_dimensions; local
699 broadcast_dimensions.push_back(i);
719 broadcast_shape, reshaped_operand, broadcast_dimensions);
    [all...]
user_computation.cc 3163 std::vector<int64> broadcast_dimensions; local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tools/parser/
hlo_parser.cc 716 optional<std::vector<int64>> broadcast_dimensions; local
718 &broadcast_dimensions};
724 shape, operands[0], *broadcast_dimensions));
    [all...]

Completed in 631 milliseconds