Searched refs:shape_partition (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emission_utils.cc | 232 std::array<int64, 3> shape_partition = in GetReductionKindAndContiguousComponentsImpl() local
234 if (shape_partition[1] == 1) { in GetReductionKindAndContiguousComponentsImpl()
236 {1, 1, shape_partition[0] * shape_partition[2]}}; in GetReductionKindAndContiguousComponentsImpl()
238 if (shape_partition[2] == 1) { in GetReductionKindAndContiguousComponentsImpl()
240 {1, shape_partition[0], shape_partition[1]}}; in GetReductionKindAndContiguousComponentsImpl()
242 return {/*is_row_reduction=*/true, shape_partition}; in GetReductionKindAndContiguousComponentsImpl()
245 std::array<int64, 3> shape_partition = in GetReductionKindAndContiguousComponentsImpl() local
248 if (shape_partition[2] == 1) { in GetReductionKindAndContiguousComponentsImpl()
250 {1, shape_partition[0], shape_partition[1]}}; in GetReductionKindAndContiguousComponentsImpl()
252 return {/*is_row_reduction=*/false, shape_partition}; in GetReductionKindAndContiguousComponentsImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | BUILD | 379 ":shape_partition",
456 ":shape_partition",
1017 name = "shape_partition",
1018 srcs = ["shape_partition.cc"],
1019 hdrs = ["shape_partition.h"],
1029 ":shape_partition",
1044 ":shape_partition",
|