Home
last modified time | relevance | path

Searched defs:axis_dim (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dcumsum_spmd_expander.cc62 StatusOr<int64_t> axis_dim = GetAxisDimension(op); in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dpack_op.cc87 const int64_t axis_dim = output_shape.dim_size(axis); in Compute() local
H A Dunpack_op.cc95 const Eigen::DenseIndex axis_dim = input_shape.dim_size(axis); in Compute() local