Searched refs:minor_starts (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 290 ArrayRef<int64_t> minor_starts, in SliceInMinorDims() argument
295 int64_t major_dims = type.getRank() - minor_starts.size(); in SliceInMinorDims()
296 std::copy(minor_starts.begin(), minor_starts.end(), in SliceInMinorDims()
327 ArrayRef<Value> minor_starts, in DynamicSliceInMinorDims() argument
330 if (minor_starts.empty()) return v; in DynamicSliceInMinorDims()
333 loc, minor_starts, type.getRank(), builder); in DynamicSliceInMinorDims()
334 int64_t major_dims = type.getRank() - minor_starts.size(); in DynamicSliceInMinorDims()
347 ArrayRef<Value> minor_starts, in DynamicUpdateSliceInMinorDims() argument
349 if (minor_starts.empty()) return v; in DynamicUpdateSliceInMinorDims()
352 loc, minor_starts, type.getRank(), builder); in DynamicUpdateSliceInMinorDims()
[all …]
|