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
272 std::vector<int64_t> output_tiling; member
|
| D | transpose.cc | 883 absl::variant<Tiling, Striding> input_layout, Tiling output_tiling, in Create() argument
1334 absl::InlinedVector<int64_t, 4> output_tiling; member
1376 TransposePlan::Tiling output_tiling, in GetOrCreate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 347 absl::InlinedVector<int64, 4> output_tiling( in PartitionIndexParallelDimensions() local
|