Searched defs:slice_index (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_split_op.cc | 84 for (int slice_index = 0; slice_index < num_split_; ++slice_index) { in Compute() local
|
| /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()
529 int slice_index = GetSliceIndex(dim, split_size, residual); in Split() local
553 const int slice_index = GetSliceIndex(dim, split_size, residual); in Split() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_morphing.cpp | 311 int slice_index = i + 3 * (j + 5 * (k + 7 * l)); in test_slice_raw_data() local
326 int slice_index = l + 11 * (k + 7 * (j + 5 * i)); in test_slice_raw_data() local
|
| /external/libgav1/libgav1/src/ |
| D | film_grain.cc | 564 const int slice_index = job_index / num_planes; in BlendNoiseChromaWorker() local
|
| /external/rust/crates/nom/src/ |
| D | traits.rs | 324 fn slice_index(&self, count: usize) -> Result<usize, Needed>; in slice_index() method
356 fn slice_index(&self, count: usize) -> Result<usize, Needed> { in slice_index() method
401 fn slice_index(&self, count: usize) -> Result<usize, Needed> { in slice_index() method
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_utils.cc | 321 int slice_index = flatten_index / slice_size; in ExtractMinMaxFromAttr() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1134 llvm_ir::IrArray::Index slice_index(multidim_index, outer_array.GetShape(), in SliceOutInnerArray() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1832 auto slice_index = SliceFirstK(index_gte, &b_, sort_dim, k.value()); in HandleSort() local
|