Searched refs:input_shape_dims (Results 1 – 2 of 2) sorted by relevance
31 PrimitiveType primitive_type, absl::Span<const int64> input_shape_dims, in CreateModuleWithProgramShape() argument34 Shape input_shape = ShapeUtil::MakeShape(primitive_type, input_shape_dims); in CreateModuleWithProgramShape()
870 absl::Span<int64 const> input_shape_dims(shape.dimensions()); in CollapseFirstNDims() local872 std::accumulate(input_shape_dims.begin(), input_shape_dims.begin() + n, in CollapseFirstNDims()876 std::copy(input_shape_dims.begin() + n, input_shape_dims.end(), in CollapseFirstNDims()