Searched defs:slice_index (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_split_op_gpu.cu.cc | 74 inline Index GetSliceShape(const Index slice_index, const Index split_size, in GetSliceShape()
139 for (int slice_index : GpuGridRangeX<int>(num_split)) { in SparseSplitFindSliceEndsKernel() local
175 int slice_index = in SparseSplitScatterKernel() local
351 for (int slice_index = 0; slice_index < num_split; ++slice_index) { in AllocateOutputs() local
|
| D | sparse_split_op.cc | 48 for (int slice_index = 0; slice_index < num_split; ++slice_index) { in operator ()() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 250 static inline int GetSliceShape(const int slice_index, const int split_size, in GetSliceShape()
530 int slice_index = GetSliceIndex(dim, split_size, residual); in Split() local
554 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/tensorflow/tensorflow/core/tpu/kernels/ |
| D | sharding_util_ops.cc | 431 int slice_index) TF_ATTRIBUTE_NOINLINE { in SliceAndMaybePadState()
740 int slice_index) TF_ATTRIBUTE_NOINLINE { in MaybeUnpadAndAssignState()
|
| /external/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 | 390 int slice_index = flatten_index / slice_size; in ExtractMinMaxFromAttr() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1313 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 | 2524 auto slice_index = SliceFirstK(index_gte, &b_, sort_dim, k.value()); in HandleSort() local
|