| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_mkldnn_contraction_kernel_test.cc | 63 Eigen::array<Index, 2> slice_start = RandomDims<Index, 2>(0, 250); in TEST() local
68 slice_start[i] = numext::mini(dims[i] - 1, slice_start[i]); in TEST()
69 slice_size[i] = numext::mini(slice_size[i], dims[i] - slice_start[i]); in TEST()
80 data_mapper.getSubMapper(slice_start[0], slice_start[1]), in TEST()
84 slice_dst = src.slice(slice_start, slice_size); in TEST()
|
| /external/rust/crates/grpcio-sys/grpc/src/core/lib/slice/ |
| D | percent_encoding.cc | 47 const uint8_t* slice_start = GRPC_SLICE_START_PTR(slice); in grpc_percent_encode_slice() local
51 for (p = slice_start; p < slice_end; p++) { in grpc_percent_encode_slice()
63 for (p = slice_start; p < slice_end; p++) { in grpc_percent_encode_slice()
|
| /external/grpc-grpc/src/core/lib/slice/ |
| D | percent_encoding.cc | 47 const uint8_t* slice_start = GRPC_SLICE_START_PTR(slice); in grpc_percent_encode_slice() local
51 for (p = slice_start; p < slice_end; p++) { in grpc_percent_encode_slice()
63 for (p = slice_start; p < slice_end; p++) { in grpc_percent_encode_slice()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 436 slice_start = slice_spec.start
437 if slice_start is None:
438 slice_start = 0 if slice_step > 0 else self._shape[0] - 1
439 elif slice_start < 0:
440 slice_start += self._shape[0]
457 cur = slice_start
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_stitch_op.cc | 172 std::vector<int64> slice_start(result_rank); in Compile() local
182 slice_start[0] = src_slice_vector[index_num]; in Compile()
187 xla::Slice(expression, slice_start, slice_limit, stride); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_loop_fusion.cc | 404 int64 slice_start = 0; in CreateFusedComputation() local
409 int64 slice_limit = slice_start + ShapeUtil::ElementsIn(shape); in CreateFusedComputation()
412 MakeSliceHlo(concated_output, {slice_start}, {slice_limit}, in CreateFusedComputation()
414 slice_start = slice_limit; in CreateFusedComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | slice_test.cc | 205 int64 slice_start; 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()
246 return absl::StrFormat("%d_%d_%d_%d", spec.input_dim0, spec.slice_start, in SliceR1TestDataToString()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_constant_folding_test.cc | 155 const int64 slice_start[] = {4, 2, 3, 1, 5}; in TEST_F() local
165 shape, literal_instruction, slice_start, slice_limits, slice_strides)); in TEST_F()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | tensorflow_op_layer_test.py | 586 slice_start = keras.Input(shape=(), dtype='int32')
590 out = inp[..., first_dim[0], slice_start[0]:slice_stop[0]:slice_stride[0]]
592 inputs=[inp, first_dim, slice_start, slice_stop, slice_stride],
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 218 llvm::SmallVector<int64_t, 8> slice_start(slice_end.size(), 0); in matchAndRewrite() local
223 slice_start[map_index] += offset; in matchAndRewrite()
230 slice_shape[i] = slice_end[i] - slice_start[i]; in matchAndRewrite()
236 GetI64ElementsAttr(slice_start, &rewriter), in matchAndRewrite()
2537 auto slice_start = (*(start.begin() + dimension)).getSExtValue(); in matchAndRewrite() local
2558 if (running_size + dim_size > slice_start && in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 658 Value slice_start = CreateI32SplatConst(loc, rewriter, {1}, 1); in CreateCondTrueBranch() local
660 loc, RankedTensorType::get({-1}, shape_dtype), input_shape, slice_start, in CreateCondTrueBranch()
|