Searched refs:slice_limit (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_stitch_op.cc | 173 std::vector<int64> slice_limit(result_rank); in Compile() local
176 slice_limit[1 + d - indices0_shape.dims()] = data0_shape.dim_size(d); in Compile()
183 slice_limit[0] = src_slice_vector[index_num] + 1; in Compile()
187 xla::Slice(expression, slice_start, slice_limit, stride); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | slice_test.cc | 206 int64 slice_limit; member
225 Slice(original, {spec.slice_start}, {spec.slice_limit}, in Run()
230 for (int i = spec.slice_start; i < spec.slice_limit; in Run()
247 spec.slice_limit, spec.slice_stride); in SliceR1TestDataToString()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_loop_fusion.cc | 409 int64 slice_limit = slice_start + ShapeUtil::ElementsIn(shape); in CreateFusedComputation() local
412 MakeSliceHlo(concated_output, {slice_start}, {slice_limit}, in CreateFusedComputation()
414 slice_start = slice_limit; in CreateFusedComputation()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 2538 auto slice_limit = (*(limit.begin() + dimension)).getSExtValue(); in matchAndRewrite() local
2565 if (running_size < slice_limit) { in matchAndRewrite()
|