| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 441 slice_end = slice_spec.stop
442 if slice_end is None:
446 slice_end = self._shape[0] if slice_step > 0 else -1
447 elif slice_end < 0:
448 slice_end += self._shape[0]
466 if cur >= var_end or cur >= slice_end:
470 end = min(slice_end, var_end) - var_start
480 if cur < var_start or cur <= slice_end:
484 if slice_end >= var_start:
485 end = slice_end - var_start
|
| /external/rust/crates/grpcio-sys/grpc/src/core/lib/slice/ |
| D | percent_encoding.cc | 48 const uint8_t* slice_end = GRPC_SLICE_END_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 | 48 const uint8_t* slice_end = GRPC_SLICE_END_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/core/kernels/data/experimental/ |
| D | rebatch_dataset_op.cc | 414 int64 slice_end = std::min(offset_ + desired_batch_size - batch_size, in GetNextInternal() local
420 slices.push_back(tensor.Slice(offset_, slice_end)); in GetNextInternal()
424 batch_size += (slice_end - offset_); in GetNextInternal()
425 offset_ = slice_end; in GetNextInternal()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 253 absl::InlinedVector<int64, 4> slice_begin, slice_end, slice_strides; in Compile() local
257 slice_end.push_back(std::max(end[i], begin[i])); in Compile()
263 slice_end.push_back(std::max(input_shape.dim_size(i) - end[i] - 1, in Compile()
272 slice = xla::Slice(slice, slice_begin, slice_end, slice_strides); in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 216 auto slice_end = in matchAndRewrite() local
218 llvm::SmallVector<int64_t, 8> slice_start(slice_end.size(), 0); in matchAndRewrite()
224 slice_end[map_index] += offset; in matchAndRewrite()
227 llvm::SmallVector<int64_t, 8> slice_stride(slice_end.size(), 1); in matchAndRewrite()
228 llvm::SmallVector<int64_t, 8> slice_shape(slice_end.size()); in matchAndRewrite()
229 for (int64_t i = 0; i < slice_end.size(); ++i) { in matchAndRewrite()
230 slice_shape[i] = slice_end[i] - slice_start[i]; in matchAndRewrite()
237 GetI64ElementsAttr(slice_end, &rewriter), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1810 const Tensor* slice_end = ic->input_tensor(2); in MaybeUpdateNodeContextOutput() local
1811 valid &= slice_end != nullptr && slice_end->NumElements() == 1; in MaybeUpdateNodeContextOutput()
1848 (slice_end->dtype() == DT_INT32 ? slice_end->flat<int32>()(0) in MaybeUpdateNodeContextOutput()
1849 : slice_end->flat<int64>()(0)); in MaybeUpdateNodeContextOutput()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1860 SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, in GetSlicedBoundRanges() argument
1886 slice_begin, slice_end, slice_stride); in GetSlicedBoundRanges()
2028 SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, in GetSlicedShapeAndBoundRanges() argument
2057 slice_begin, slice_end, slice_stride); in GetSlicedShapeAndBoundRanges()
|
| D | tf_generated_ops.td | 15406 ::llvm::SmallVectorImpl<int64_t> *slice_end,
15456 ::llvm::SmallVectorImpl<int64_t> *slice_end,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3339 NodeDef* slice_end; in GetStridedSliceAxis() local
3342 TF_RETURN_IF_ERROR(GetInputNode(node->input(2), &slice_end)); in GetStridedSliceAxis()
3345 for (const auto* n : {slice_begin, slice_end, slice_strides}) { in GetStridedSliceAxis()
3357 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_end, "value")); in GetStridedSliceAxis()
3358 if (!slice_end_t.FromProto(slice_end->attr().at("value").tensor())) { in GetStridedSliceAxis()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 1252 int64 slice_end = operands[i]->slice_limits(concatenate_dimension); in HandleConcatenate() local
1258 operands[j]->slice_starts(concatenate_dimension) == slice_end) { in HandleConcatenate()
1276 slice_end = operands[j]->slice_limits(concatenate_dimension); in HandleConcatenate()
1283 slice_end - operands[i]->slice_starts(concatenate_dimension)); in HandleConcatenate()
1286 new_limit_indices[concatenate_dimension] = slice_end; in HandleConcatenate()
|