Searched defs:processing_dims (Results 1 – 2 of 2) sorted by relevance
87 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); in HandleStridedSliceCase() local123 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); in HandleStridedSliceGradCase() local146 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); in operator() local
138 const int processing_dims = processing_shape.dims(); in Compute() local243 const int processing_dims = processing_shape.dims(); in Compute() local342 const int processing_dims = processing_shape.dims(); in Compute() local