Home
last modified time | relevance | path

Searched defs:offset_dims (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc253 HloInstruction* accumulator, absl::Span<const int64> offset_dims, in PermuteBatchAndOffsetDims()
Dhlo_instructions.cc2323 string offset_dims = in GatherDimensionNumbersToString() local
2341 absl::Span<const int64> offset_dims, in MakeGatherDimNumbers()
Dhlo_parser.cc1615 optional<std::vector<int64>> offset_dims; in ParseInstructionRhs() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto452 repeated int64 offset_dims = 1; field
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3398 std::vector<int64> offset_dims = {x_dims[is_nhwc ? 3 : 1]}; in TEST_F() local