| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/ |
| H A D | DimUtils.h | 73 inline int32_t dim_at(const std::vector<int64_t>& sizes, DimIndex dim_index) { in dim_at()
91 inline int32_t dim_at(const api::vTensor& v_in, DimIndex dim_index) { in dim_at()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Split.cpp | 32 DimIndex dim_index = normalize_to_dim_index(*t_in, dim); in add_split_with_sizes_default_node() local
125 DimIndex dim_index = normalize_to_dim_index(*t_in, dim); in add_split_tensor_node() local
|
| H A D | Cat.cpp | 34 DimIndex dim_index = normalize_to_dim_index(*t_out, dim); in add_cat_default_node() local
|
| H A D | Slice.cpp | 62 DimIndex dim_index = normalize_to_dim_index(*t_in, dim); in add_slice_tensor_copy_node() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_padder.cc | 377 HloInstruction* dim_index = in GenerateBinaryMask() local
912 for (int64_t dim_index = 0; dim_index < input_window->dimensions_size(); in RewriteInputWithDynamicPadding() local
1186 for (int64_t dim_index = 0; dim_index < rank; ++dim_index) { in RewriteDynamicReduceWindowSamePadding() local
1229 for (int64_t dim_index = 0; dim_index < rank; ++dim_index) { in RewriteDynamicSelectAndScatterSamePadding() local
1274 for (int64_t dim_index = 0; dim_index < rank; ++dim_index) { in RewriteDynamicSelectAndScatterSamePadding() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | tensor.cpp | 46 size_t dim_index = reduce_ndim - i - 1; in constructStmt() local
92 for (auto dim_index : loop_order) { in constructStmt() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | shape_op.cc | 132 int64_t dim_index; in Compile() local
161 int64_t dim_index; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | c_api_debug.cc | 82 TFE_TensorDebugInfo* debug_info, int dim_index) { in TFE_TensorDebugInfoOnDeviceDim()
|
| H A D | custom_device_testutil.cc | 48 int64_t LoggedTensorDim(void* data, int dim_index, TF_Status* status) { in LoggedTensorDim()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | literal.cc | 338 int32_t LiteralBase::GetDynamicSize(int64_t dim_index, in GetDynamicSize()
579 void LiteralBase::Piece::SetDynamicSize(int64_t dim_index, int32_t size) { in SetDynamicSize()
669 void MutableLiteralBase::SetDynamicSize(int64_t dim_index, int32_t size) { in SetDynamicSize()
673 void MutableLiteralBase::SetDynamicSize(int64_t dim_index, in SetDynamicSize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor.cc | 1089 void PrintOneDim(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDim()
1127 void PrintDimSpacing(int dim_index, int num_dims, string* result) { in PrintDimSpacing()
1142 void PrintOneDimV2(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDimV2()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | tensor_handle_data.cc | 58 Status LocalTensorHandleData::Dim(int dim_index, int64_t* dim) const { in Dim()
|
| H A D | custom_device_test.cc | 79 Status Dim(int dim_index, int64_t* dim) const override { in Dim()
|
| H A D | tensor_handle.cc | 91 Status TensorHandle::PackedTensorHandleData::Dim(int dim_index, in Dim()
638 Status TensorHandle::Dim(int dim_index, int64_t* dim) const { in Dim()
|
| /aosp_15_r20/external/skia/modules/skottie/src/effects/ |
| H A D | GaussianBlurEffect.cpp | 69 const auto dim_index = SkTPin<size_t>(static_cast<size_t>(fDimensions), in onSync() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | c_api_opaque.cc | 30 int32_t dim_index) { in TfLiteOpaqueTensorDim()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | reduction_ops_common.cc | 111 int dim_index = 0; in Simplify() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | elementwise_spmd_expander.cc | 168 const int dim_index = dim_shape_and_index.index(); in ComputeLayoutBackward() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | helpers.cpp | 41 int64_t dim_index = dim < 0 ? rank + dim : dim; in GetCanonicalDimensionIndex() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | attribute_importer.cc | 192 for (int64_t dim_index : xla_layout.minor_to_major()) in ExtractLayoutsFromShapes() local
|
| /aosp_15_r20/external/angle/third_party/spirv-tools/src/source/val/ |
| H A D | validate_tensor_layout.cpp | 102 const auto dim_index = 1; in ValidateTensorTypeWithDimValuesNV() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/ |
| H A D | eager_operation_jni.cc | 135 jint dim_index) { in Java_org_tensorflow_EagerOperation_dim()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | remote_tensor_handle_data.cc | 143 Status RemoteTensorHandleData::Dim(int dim_index, int64_t* dim) const { in Dim()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | reduce_util.h | 61 size_t dim_index[kTensorDimensionLimit]; in apply_on_flat_ix_with_dim_mask_and_base() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/ |
| H A D | parallel_device.cc | 222 int64_t ParallelTensorDim(void* data, int dim_index, TF_Status* status) { in ParallelTensorDim()
|