Searched defs:new_dimensions (Results 1 – 7 of 7) sorted by relevance
368 std::vector<int64_t> new_dimensions = in HandleReverse() local444 std::vector<int64_t> new_dimensions; in TransformDimensionsForLayoutPreservingHlo() local
747 std::vector<int64_t> new_dimensions(activations->shape().dimensions().begin(), in SplitAndTransposeMergedBatch() local800 std::vector<int64_t> new_dimensions(activations->shape().dimensions().begin(), in ChangeSpatialSizeOnSpaceToBatchedShape() local1814 std::vector<int64_t> new_dimensions( in Propagate() local1968 std::vector<int64_t> new_dimensions(consumer->dimensions().size()); in Propagate() local2402 std::vector<int64_t> new_dimensions(new_instr->shape().dimensions().begin(), in BatchToSpace() local
413 std::vector<int64_t> new_dimensions(arg_shape->dimensions().begin(), in InferConcatOpShape() local2157 std::vector<int64_t> new_dimensions(shape.dimensions().begin(), in InferAllToAllShape() local2269 std::vector<int64_t> new_dimensions; in InferReduceShape() local
3546 std::vector<int64_t> new_dimensions; in HandleBroadcast() local5431 std::vector<int64_t> new_dimensions; in HandleReduce() local
103 int larger_dimensions, new_dimensions; in vp9_lookahead_push() local
110 int larger_dimensions, new_dimensions; in av1_lookahead_push() local
389 absl::Span<const int64_t> new_dimensions, in ReshapeSlice()