| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_split_op.cc | 76 for (int slice_index = 0; slice_index < num_split_; ++slice_index) { in Compute() local
77 context->set_output(slice_index, outputs[slice_index].indices()); in Compute()
78 context->set_output(slice_index + num_split_, in Compute()
79 outputs[slice_index].values()); in Compute()
82 slice_index + 2 * num_split_, in Compute()
83 {outputs[slice_index].dims()}, &shape)); in Compute()
84 auto output_shape = outputs[slice_index].shape(); in Compute()
85 for (int dim = 0; dim < outputs[slice_index].dims(); ++dim) { in Compute()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 249 static inline int GetSliceShape(const int slice_index, const int split_size, in GetSliceShape() argument
252 DCHECK_GE(slice_index, 0); in GetSliceShape()
254 if (slice_index < residual) { in GetSliceShape()
529 int slice_index = GetSliceIndex(dim, split_size, residual); in Split() local
530 num_values[slice_index]++; in Split()
549 const int slice_index = GetSliceIndex(dim, split_size, residual); in Split() local
550 const int slice_dim = values_inserted_in_slice[slice_index]++; in Split()
551 output_values_t[slice_index](slice_dim) = input_values_t(i); in Split()
554 output_indices_t[slice_index](slice_dim, j) = in Split()
|
| /external/rust/crates/nom/src/bytes/ |
| D | complete.rs | 257 let res: IResult<_, _, Error> = if let Some(index) = input.slice_index(idx) { in take_while_m_n()
264 let res: IResult<_, _, Error> = if let Some(index) = input.slice_index(n) { in take_while_m_n()
279 match input.slice_index(n) { in take_while_m_n()
384 move |i: Input| match i.slice_index(c) { in take()
|
| D | streaming.rs | 263 let res: IResult<_, _, Error> = if let Some(index) = input.slice_index(idx) { in take_while_m_n()
270 let res: IResult<_, _, Error> = if let Some(index) = input.slice_index(n) { in take_while_m_n()
285 match input.slice_index(n) { in take_while_m_n()
390 move |i: Input| match i.slice_index(c) { in take()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_morphing.cpp | 289 int slice_index = i + 3 * (j + 5 * (k + 7 * l)); in test_slice_raw_data() local
290 VERIFY_IS_EQUAL(slice5.data()[slice_index], tensor(i,j,k,l+4)); in test_slice_raw_data()
304 int slice_index = l + 11 * (k + 7 * (j + 5 * i)); in test_slice_raw_data() local
305 VERIFY_IS_EQUAL(slice5.data()[slice_index], tensor(i+1,j,k,l)); in test_slice_raw_data()
|
| /external/libgav1/libgav1/src/ |
| D | film_grain.cc | 553 const int slice_index = job_index / num_planes; in BlendNoiseChromaWorker() local
554 const int start_height = slice_index * kFrameChunkHeight; in BlendNoiseChromaWorker()
|
| /external/rust/crates/nom/src/ |
| D | traits.rs | 323 fn slice_index(&self, count: usize) -> Option<usize>; in slice_index() method
359 fn slice_index(&self, count: usize) -> Option<usize> { in slice_index() method
404 fn slice_index(&self, count: usize) -> Option<usize> { in slice_index() method
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_utils.cc | 301 int slice_index = flatten_index / slice_size; in ExtractMinMaxFromAttr() local
302 int channel_index = slice_index % dim_size; in ExtractMinMaxFromAttr()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1115 llvm_ir::IrArray::Index slice_index(multidim_index, outer_array.GetShape(), in SliceOutInnerArray() local
1117 llvm::Value* slice_ptr = outer_array.EmitArrayElementAddress(slice_index, b); in SliceOutInnerArray()
|
| /external/python/cpython3/Lib/test/ |
| D | clinic.test | 1452 a: slice_index = 12
1453 b: slice_index(accept={int}) = 34
1454 c: slice_index(accept={int, NoneType}) = 56
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1715 auto slice_index = SliceFirstK(index_gte, &b_, sort_dim, k.value()); in HandleSort() local
1716 slice_index->set_sharding(input_sharding); in HandleSort()
1718 slice_index, ShapeUtil::MakeShape(index_type, replicated_dimensions), in HandleSort()
|