Searched refs:limit_index (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2237 int64 limit_index = limits[dimension]; in InferSliceShape() local
2242 if (limit_index > arg.dimensions(dimension)) { in InferSliceShape()
2246 limit_index, arg.dimensions(dimension))); in InferSliceShape()
2249 VLOG(2) << StrFormat("limits[%d] = %d", dimension, limit_index); in InferSliceShape()
2250 if (start_index > limit_index) { in InferSliceShape()
2254 limit_index, start_index)); in InferSliceShape()
2259 sizes.push_back((limit_index - start_index + stride - 1) / stride); in InferSliceShape()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 523 int64 start_index, int64 limit_index, in SliceInDim() argument
525 return xla::SliceInDim(operand.op(), start_index, limit_index, stride, dimno); in SliceInDim()
|
| D | local_computation_builder.h | 267 int64 limit_index, int64 stride, int64 dimno);
|
| D | xla_client.py | 1337 def SliceInDim(self, operand, start_index, limit_index, stride, dimno): argument
1350 return self._client.SliceInDim(operand, start_index, limit_index, stride,
|
| D | xla_client_test.py | 1058 limit_index=2,
1065 limit_index=3,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 350 XlaOp SliceInDim(const XlaOp& operand, int64 start_index, int64 limit_index,
739 int64 limit_index, int64 stride, int64 dimno);
1201 XlaOp SliceInDim(const XlaOp& operand, int64 start_index, int64 limit_index,
|
| D | xla_builder.cc | 744 int64 limit_index, int64 stride, int64 dimno) { in SliceInDim() argument
752 limits[dimno] = limit_index; in SliceInDim()
2875 XlaOp SliceInDim(const XlaOp& operand, int64 start_index, int64 limit_index, in SliceInDim() argument
2877 return operand.builder()->SliceInDim(operand, start_index, limit_index, in SliceInDim()
|