Searched defs:output_tiling (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | transpose_test.cc | 50 absl::InlinedVector<int64_t, 4> output_tiling = {1, 1, 1, 1, 1, 1, 1}; in TEST() local
74 absl::InlinedVector<int64_t, 4> output_tiling = {1, 1, 1, 1, 1, 1, 1}; in TEST() local
271 std::vector<int64_t> output_tiling; member
|
| D | transpose.cc | 883 std::variant<Tiling, Striding> input_layout, Tiling output_tiling, in Create() argument
1334 absl::InlinedVector<int64_t, 4> output_tiling; member
1369 TransposePlan::Tiling output_tiling, in GetOrCreate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 513 absl::InlinedVector<int64_t, 4> output_tiling( in PartitionGatherIndexParallelDimensions() local
|