Searched refs:limit_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2436 int64 limit_index = limits[dimension]; in InferSliceShape() local
2441 if (limit_index > arg.dimensions(dimension)) { in InferSliceShape()
2445 limit_index, arg.dimensions(dimension))); in InferSliceShape()
2448 VLOG(2) << StrFormat("limits[%d] = %d", dimension, limit_index); in InferSliceShape()
2449 if (start_index > limit_index) { in InferSliceShape()
2453 limit_index, start_index)); in InferSliceShape()
2458 sizes.push_back((limit_index - start_index + stride - 1) / stride); in InferSliceShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 491 virtual XlaOp SliceInDim(XlaOp operand, int64 start_index, int64 limit_index,
1094 friend XlaOp SliceInDim(XlaOp operand, int64 start_index, int64 limit_index,
1755 XlaOp SliceInDim(XlaOp operand, int64 start_index, int64 limit_index,
|
| D | xla_builder.cc | 980 int64 limit_index, int64 stride, int64 dimno) { in SliceInDim() argument
988 limits[dimno] = limit_index; in SliceInDim()
4161 XlaOp SliceInDim(const XlaOp operand, int64 start_index, int64 limit_index, in SliceInDim() argument
4163 return operand.builder()->SliceInDim(operand, start_index, limit_index, in SliceInDim()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1140 limit_index=2,
1147 limit_index=3,
|