Home
last modified time | relevance | path

Searched defs:lhs_batch_dims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc272 absl::Span<const int64_t> lhs_batch_dims = in AddBackendConstraints() local
Dmatmul_utils.cc234 const Shape& lhs_shape, absl::Span<const int64_t> lhs_batch_dims, in For()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc2751 optional<std::vector<int64_t>> lhs_batch_dims; in CreateInstruction() local