Searched refs:SliceInternal (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 714 Literal LiteralBase::SliceInternal( in SliceInternal() function in xla::LiteralBase
747 return SliceInternal<bool>(result_shape, start_indices); in Slice()
749 return SliceInternal<uint8>(result_shape, start_indices); in Slice()
751 return SliceInternal<uint16>(result_shape, start_indices); in Slice()
753 return SliceInternal<uint32>(result_shape, start_indices); in Slice()
755 return SliceInternal<uint64>(result_shape, start_indices); in Slice()
757 return SliceInternal<int8>(result_shape, start_indices); in Slice()
759 return SliceInternal<int16>(result_shape, start_indices); in Slice()
761 return SliceInternal<int32>(result_shape, start_indices); in Slice()
763 return SliceInternal<int64>(result_shape, start_indices); in Slice()
[all …]
|
| D | literal.h | 545 Literal SliceInternal(const Shape& result_shape,
|