Searched refs:slice_index (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_split_op.cc | 77 for (int slice_index = 0; slice_index < num_split_; ++slice_index) { in Compute() local
78 context->set_output(slice_index, outputs[slice_index].indices()); in Compute()
79 context->set_output(slice_index + num_split_, in Compute()
80 outputs[slice_index].values()); in Compute()
83 slice_index + 2 * num_split_, in Compute()
84 {outputs[slice_index].dims()}, &shape)); in Compute()
85 auto output_shape = outputs[slice_index].shape(); in Compute()
86 for (int dim = 0; dim < outputs[slice_index].dims(); ++dim) { in Compute()
|
| /external/u-boot/arch/arm/mach-imx/mx8m/ |
| D | clock_slice.c | 494 u32 slice_index) in get_clk_root_target() argument
501 (void __iomem *)&ccm_reg->core_root[slice_index]; in get_clk_root_target()
505 (void __iomem *)&ccm_reg->bus_root[slice_index]; in get_clk_root_target()
509 (void __iomem *)&ccm_reg->ip_root[slice_index]; in get_clk_root_target()
513 (void __iomem *)&ccm_reg->ahb_ipg_root[slice_index * 2]; in get_clk_root_target()
517 (void __iomem *)&ccm_reg->ahb_ipg_root[slice_index * 2 + 1]; in get_clk_root_target()
546 clk_root_target = get_clk_root_target(p->slice_type, p->slice_index); in clock_get_target_val()
569 clk_root_target = get_clk_root_target(p->slice_type, p->slice_index); in clock_set_target_val()
581 u32 slice_index, slice_type; in clock_root_enabled() local
593 slice_index = root_array[root_entry].slice_index; in clock_root_enabled()
[all …]
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 357 static inline int GetSliceShape(const int slice_index, const int split_size, in GetSliceShape() argument
360 DCHECK_GE(slice_index, 0); in GetSliceShape()
362 if (slice_index < residual) { in GetSliceShape()
608 int slice_index = GetSliceIndex(dim, split_size, residual); in Split() local
609 num_values[slice_index]++; in Split()
628 const int slice_index = GetSliceIndex(dim, split_size, residual); in Split() local
629 const int slice_dim = values_inserted_in_slice[slice_index]++; in Split()
630 output_values_t[slice_index](slice_dim) = input_values_t(i); in Split()
633 output_indices_t[slice_index](slice_dim, j) = in Split()
|
| /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/u-boot/arch/arm/include/asm/arch-mx8m/ |
| D | clock.h | 408 u32 slice_index; member
|
| /external/python/cpython3/Lib/test/ |
| D | clinic.test | 781 a: slice_index = 12
782 b: slice_index(accept={int}) = 34
783 c: slice_index(accept={int, NoneType}) = 56
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 919 llvm_ir::IrArray::Index slice_index(multidim_index, outer_array.GetShape(), in SliceOutInnerArray() local
921 llvm::Value* slice_ptr = outer_array.EmitArrayElementAddress(slice_index, b); in SliceOutInnerArray()
|