Home
last modified time | relevance | path

Searched defs:last_dim (Results 1 – 25 of 28) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dnn_ops.cc114 DimensionHandle last_dim = c->Dim(input, 3); in __anon84cbdd650202() local
148 DimensionHandle last_dim = c->Dim(input, 3); in __anon84cbdd650302() local
1344 DimensionHandle last_dim = c->Dim(input, -1); in TopKShapeFn() local
1495 DimensionHandle last_dim = c->Dim(input, -1); in __anon84cbdd652102() local
1737 DimensionHandle last_dim = c->Dim(input, 3); in __anon84cbdd652902() local
H A Dimage_ops.cc136 DimensionHandle last_dim; in ColorspaceShapeFn() local
H A Darray_ops.cc3343 ShapeHandle last_dim = c->Vector(c->Dim(inputs, -1)); in __anon38bbb0e84b02() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dsorting.cc32 int last_dim = input_shape.dimensions_size() - 1; in TopK() local
157 int last_dim = input_shape.dimensions_size() - 1; in TopKWithPartitions() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/
H A DSpectralOps.cpp180 auto last_dim = dim.back(); in _fft_r2c_mkl_out() local
284 auto last_dim = dim.back(); in _fft_r2c_mkl() local
544 auto last_dim = dim.back(); in _fft_r2c_mkl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSpectralOps.cpp335 auto last_dim = dim.back(); in _fft_r2c_cufft() local
409 auto last_dim = dim.back(); in _fft_r2c_cufft_out() local
H A DUnfoldBackwardKernel.cu130 auto last_dim = maybe_wrap_dim(-1, grad_in.dim()); in unfold_backward_cuda_kernel() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtopk_op.cc35 int last_dim = input_shape.dims() - 1; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/c/kernels/ops/
H A Dbitcast.cc43 TF_DimensionHandle* last_dim = TF_NewDimensionHandle(); in ComputeNewShape() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_bias_add_op.cc73 const auto last_dim = input.shape().dims() - 1; in Compute() local
H A Dnth_element_op.cc95 const int last_dim = input_tensor.dim_size(input_tensor.dims() - 1); in operator ()() local
H A Dquantized_batch_norm_op.cc241 const auto last_dim = input.shape().dims() - 1; in Compute() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUnfoldBackwardKernel.cpp120 auto last_dim = maybe_wrap_dim(-1, grad_in.dim()); in unfold_backward_cpu_kernel() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_tile_transpose.cc135 unsigned last_dim = num_loops - 1; in runOnOperation() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqsoftmax.cpp46 const int64_t last_dim = qx.dim() - 1; in qsoftmax_qnnpack() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DWeightNorm.cpp131 int64_t last_dim = saved_v.dim() - 1; in _weight_norm_differentiable_backward() local
H A DSpectralOps.cpp516 const auto last_dim = desc.dim.back(); in canonicalize_fft_c2r_shape_and_dim_args() local
580 const auto last_dim = desc.dim.back(); in fft_hfftn_impl() local
616 const auto last_dim = desc.dim.back(); in fft_ihfftn_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dmath_kernel_test.cpp117 auto test_bmm = [](int64_t last_dim) { in TEST()
/aosp_15_r20/external/executorch/backends/cadence/reference/operators/
H A Dquantized_layer_norm.cpp45 size_t last_dim = input.size(input.dim() - 1); in quantized_layer_norm_per_tensor_() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dprepare_tf.cc992 int64_t last_dim = -1; in matchAndRewrite() local
994 auto is_last_dim_compatible = [](const Value &v, int64_t &last_dim) { in matchAndRewrite()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorUtils.cpp61 std::optional<int64_t> last_dim = nt.opt_size(-1); in get_consistent_last_dim_of_nested_tensor() local
H A DNestedTensorTransformerFunctions.cpp38 const auto last_dim = get_consistent_last_dim_of_nested_tensor(*nt_input); in check_nested_tensor_matrix_constraints() local
/aosp_15_r20/external/executorch/backends/cadence/hifi/operators/
H A Dquantized_layer_norm.cpp48 size_t last_dim = input.size(input.dim() - 1); in quantized_layer_norm_per_tensor_() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/
H A DTensor.cpp33 int64_t last_dim = int64_t(ndim) - 1 - packed_dim; in calculate_dim_order() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc2104 int last_dim = (*dims)[dims_size - 1]; in ProcessMatrixDiagOperator() local

12