Searched refs:SliceInternal (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 889 Literal LiteralBase::SliceInternal( in SliceInternal() function in xla::LiteralBase
931 return SliceInternal<bool>(result_shape, start_indices); in Slice()
933 return SliceInternal<uint8>(result_shape, start_indices); in Slice()
935 return SliceInternal<uint16>(result_shape, start_indices); in Slice()
937 return SliceInternal<uint32>(result_shape, start_indices); in Slice()
939 return SliceInternal<uint64>(result_shape, start_indices); in Slice()
941 return SliceInternal<int8>(result_shape, start_indices); in Slice()
943 return SliceInternal<int16>(result_shape, start_indices); in Slice()
945 return SliceInternal<int32>(result_shape, start_indices); in Slice()
947 return SliceInternal<int64>(result_shape, start_indices); in Slice()
[all …]
|
| D | literal.h | 579 Literal SliceInternal(const Shape& result_shape,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 241 StatusOr<XlaOp> SliceInternal(const Shape& shape, XlaOp operand,
|
| D | mlir_hlo_builder.cc | 508 StatusOr<XlaOp> MlirHloBuilder::SliceInternal( in SliceInternal() function in xla::MlirHloBuilder
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 960 return SliceInternal(shape, operand, start_indices, limit_indices, strides); in Slice()
964 StatusOr<XlaOp> XlaBuilder::SliceInternal(const Shape& shape, XlaOp operand, in SliceInternal() function in xla::XlaBuilder
|
| D | xla_builder.h | 487 virtual StatusOr<XlaOp> SliceInternal(const Shape& shape, XlaOp operand,
|